Add application file

app.py

@@ -1,7 +1,20 @@
 import gradio as gr
+from diffusers import StableDiffusionPipeline
+import torch
 
-def greet(name):
-    return "Hello " + name + "!!"
+model_id = "runwayml/stable-diffusion-v1-5"
+pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
+# pipe = pipe.to("cuda")
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
+prompt = "a photo of an astronaut riding a horse on mars"
+    
+# image.save("astronaut_rides_horse.png")
+
+
+def infer(prompt):
+    image = pipe(prompt).images[0]  
+    
+    return image
+
+iface = gr.Interface(fn=infer, inputs=gr.Textbox(lines=1, placeholder="Prompt Here..."), outputs="image")
 iface.launch()