ultimate

app.py

@@ -8,23 +8,16 @@ pipe = StableDiffusionPipeline.from_pretrained(
     revision="fp16" if torch.cuda.is_available() else "main"
 ).to("cuda" if torch.cuda.is_available() else "cpu")
 
-def generar_imagen(prompt, estilo):
-    estilos = {
-        "Médico": "representado con iluminación clínica y ambiente hospitalario",
-        "Ciberpunk": "estilo ciberpunk con luces de neón, distópico, futurista",
-        "Anime": "estilo anime, colores vibrantes, personajes kawaii",
-    }
-    prompt_completo = f"{prompt}, {estilos.get(estilo, '')}"
-    image = pipe(prompt_completo).images[0]
+def generar_imagen(prompt):
+    image = pipe(prompt).images[0]
     return image
 
 demo = gr.Interface(
     fn=generar_imagen,
-    inputs=[
-        gr.Textbox(label="Tu idea"),
-        gr.Dropdown(choices=["Médico", "Ciberpunk", "Anime"], label="Estilo de Arte")
-    ],
-    outputs="image"
+    inputs=gr.Textbox(label="Descripción de la imagen"),
+    outputs="image",
+    title="Generador de Imágenes - DRDELATV",
+    description="Escribe una descripción y genera una imagen con IA usando Stable Diffusion"
 )
 
 demo.launch()