Update app.py

app.py

@@ -96,7 +96,7 @@ def save_image(image, file_name, prompt=None):
             save_prompt(f"{file_name}: {prompt}")
         return image_path
 
-async def generate_image(prompt, width, height, seed, model_name="XLabs-AI/flux-RealismLora"):
+async def generate_image(prompt, width, height, seed, model_name):
     if seed == -1:
         seed = random.randint(0, MAX_SEED)
     image = await client.text_to_image(prompt=prompt, height=height, width=width, model=model_name)
@@ -145,6 +145,7 @@ async def main():
     prompt = st.sidebar.text_area("Descripción de la imagen", height=150, max_chars=500)
     format_option = st.sidebar.selectbox("Formato", ["9:16", "16:9", "1:1"]) 
     prompt_checkbox = st.sidebar.checkbox("Mejorar Prompt")
+    model_option = st.sidebar.selectbox("Modelo", ["black-forest-labs/FLUX.1-schnell", "black-forest-labs/FLUX.1-dev"])
     style_option = st.sidebar.selectbox("Estilo", ["realista", "hiperrealista", "photorealista", "imaginativo", "abstracto", "moebius"])
     
     width, height = (360, 640) if format_option == "9:16" else (640, 360) if format_option == "16:9" else (640, 640)
@@ -163,7 +164,7 @@ async def main():
     if st.sidebar.button("Generar Imágenes"):
         with st.spinner("Generando imágenes..."):
             try:
-                results = await gen(prompts, width, height, "XLabs-AI/flux-RealismLora", num_variants, prompt_checkbox)
+                results = await gen(prompts, width, height, model_option, num_variants, prompt_checkbox)
                 st.session_state['generated_image_paths'] = results
                 for result in results:
                     st.image(result, caption="Imagen Generada")
@@ -174,4 +175,4 @@ async def main():
 
 
 if __name__ == "__main__":
-    asyncio.run(main())
+    asyncio.run(main())