Update app.py

app.py

@@ -1,8 +1,12 @@
 import torch
 from diffusers import StableDiffusionPipeline
+from PIL import Image
 
 pipe = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16)
 pipe = pipe.to("cuda")
 
-prompt = "a photo of an astronaut riding a horse on mars"
-image = pipe(prompt).images[0]
+prompt = input("Enter a prompt: ")
+image = pipe(prompt).images[0]
+
+image = Image.fromarray(image)
+image.show()