Update app.py

app.py

@@ -1,17 +1,27 @@
 import gradio as gr
 from diffusers import StableDiffusionPipeline
+import torch
+
+pipe = StableDiffusionPipeline.from_pretrained(
+    "stabilityai/sd-turbo",
+    torch_dtype=torch.float32
+)
 
-pipe = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5")
 pipe = pipe.to("cpu")
 
 def generar_imagen(prompt):
-    imagen = pipe(prompt).images[0]
+    imagen = pipe(prompt, num_inference_steps=1).images[0]
     return imagen
 
-interfaz = gr.Interface(
-    fn=generar_imagen,
-    inputs=gr.Textbox(label="Escribe tu idea"),
-    outputs=gr.Image(label="Imagen generada")
-)
+with gr.Blocks() as demo:
+    gr.Markdown("# 🎨 ImaginaAI")
+    gr.Markdown("Genera imágenes con inteligencia artificial")
+
+    prompt = gr.Textbox(label="Escribe tu idea")
+
+    boton_generar = gr.Button("Generar imagen")
+    resultado = gr.Image(label="Imagen generada")
+
+    boton_generar.click(fn=generar_imagen, inputs=prompt, outputs=resultado)
 
-interfaz.launch()
+demo.launch()