Actualizando con modelos populares y confiables de Hugging Face

app.py

@@ -30,14 +30,15 @@ MODELS = {
     },
     "image": {
         "CompVis/stable-diffusion-v1-4": "Stable Diffusion v1.4 (Básico)",
-        "runwayml/stable-diffusion-v1-5": "Stable Diffusion v1.5",
         "stabilityai/stable-diffusion-2-1": "Stable Diffusion 2.1",
         "stabilityai/stable-diffusion-xl-base-1.0": "SDXL Base",
-        "stabilityai/stable-diffusion-xl-refiner-1.0": "SDXL Refiner",
-        "prompthero/openjourney": "Midjourney style",
-        "dreamlike-art/dreamlike-photoreal-2.0": "Fotorealista",
-        "nitrosocke/Ghibli-Diffusion": "Estilo Studio Ghibli",
-        "nitrosocke/mo-di-diffusion": "Estilo moderno"
+        "stabilityai/stable-diffusion-3-medium": "SD 3 Medium",
+        "stabilityai/stable-diffusion-3.5-large": "SD 3.5 Large",
+        "black-forest-labs/FLUX.1-dev": "FLUX.1 Dev",
+        "black-forest-labs/FLUX.1-schnell": "FLUX.1 Schnell",
+        "prompthero/openjourney": "Midjourney Style",
+        "WarriorMama777/OrangeMixs": "Orange Mixs",
+        "hakurei/waifu-diffusion": "Waifu Diffusion"
     },
     "chat": {
         "microsoft/DialoGPT-medium": "Chat conversacional",
@@ -325,7 +326,7 @@ with gr.Blocks(title="Modelos Libres de IA", theme=gr.themes.Soft()) as demo:
                 with gr.Column():
                     image_model = gr.Dropdown(
                         choices=list(MODELS["image"].keys()),
-                        value="CompVis/stable-diffusion-v1-4",
+                        value="black-forest-labs/FLUX.1-schnell",
                         label="Modelo de Imagen"
                     )
                     image_prompt = gr.Textbox(

requirements.txt

@@ -5,4 +5,6 @@ diffusers>=0.20.0
 accelerate>=0.20.0
 Pillow>=9.0.0
 numpy>=1.21.0
-requests>=2.28.0 
+requests>=2.28.0
+safetensors>=0.3.0
+xformers>=0.0.20