Update app.py

app.py

@@ -1,15 +1,15 @@
 import gradio as gr
-from diffusers import DiffusionPipeline
+from diffusers import StableDiffusionPipeline
 import torch
 
+model_id = "runwayml/stable-diffusion-v1-5"
+pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
 
-pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float32)
-pipe.enable_model_cpu_offload()
 
 def infer(prompt):
 
-    prompt = "A cinematic shot of a baby racoon wearing an intricate italian priest robe."
-    image = pipe(prompt=prompt, num_inference_steps=1, guidance_scale=0.0).images[0]
+    prompt = "a photo of an astronaut riding a horse on mars"
+    image = pipe(prompt).images[0]  
 
     return image