Update app.py

app.py

@@ -1,6 +1,6 @@
 from diffusers import DiffusionPipeline
+import torch
 
-pipe = DiffusionPipeline.from_pretrained("prompthero/openjourney-v4")
-
-prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
-image = pipe(prompt).images[0]
+pipeline = DiffusionPipeline.from_pretrained("stable-diffusion-v1-5/stable-diffusion-v1-5", torch_dtype=torch.float16)
+pipeline.to("cpu")
+pipeline("An image of a squirrel in Picasso style").images[0]