Update app.py

app.py

@@ -1,19 +1,19 @@
-import spaces
+import gradio as gr
 from diffusers import DiffusionPipeline
 
-pipe = DiffusionPipeline.from_pretrained("stable-diffusion-v1-5/stable-diffusion-v1-5")
+# Cargar el modelo
+pipe = DiffusionPipeline.from_pretrained("stable-diffusion-v1-5")
 pipe.load_lora_weights("ovieyra21/yebama-sd15")
 pipe.to('cuda')
 
-@spaces.GPU
-prompt = "yebama"
-image = pipe(prompt).images[0]
 def generate(prompt):
-    return pipe(prompt).images
+    # Generar la imagen
+    image = pipe(prompt).images[0]
+    return image
 
+# Crear la interfaz de Gradio
 gr.Interface(
     fn=generate,
-    inputs=gr.Text(),
-    outputs=gr.Gallery(),
+    inputs=gr.Textbox(label="Escribe tu prompt"),
+    outputs=gr.Image(type="pil"),
 ).launch()
-