Update app.py

app.py

@@ -1,11 +1,16 @@
-from diffusers import StableDiffusionPipeline
+from diffusers import StableDiffusionPipeline as SD
 import torch
+import gradio as gr
 
 model_id = "runwayml/stable-diffusion-v1-5"
-pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16, revision="fp16")
-pipe = pipe.to("cuda")
+gen = SD.from_pretrained(model_id)
 
-prompt = "a photo of an astronaut riding a horse on mars"
-image = pipe(prompt).images[0]  
-    
-image.save("astronaut_rides_horse.png")
+gen.to("cuda")
+
+def generate():
+    prompt = "A doggo"
+    image = gen(prompt).images[0]
+    image.save("result.png")
+
+ui = gr.Interface(fn=generate, inputs="prompt_field", outputs="result")
+ui.launch()