Update app.py

app.py

@@ -1,3 +1,19 @@
-import gradio as gr
+import spaces
+from diffusers import DiffusionPipeline
+
+pipe = DiffusionPipeline.from_pretrained("stable-diffusion-v1-5/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
+
+gr.Interface(
+    fn=generate,
+    inputs=gr.Text(),
+    outputs=gr.Gallery(),
+).launch()
 
-gr.load("models/ovieyra21/yebama-sd15").launch()