Update app.py

app.py

@@ -5,7 +5,7 @@ from diffusers import StableDiffusionPipeline
 
 def generate_image(input_text):
     # Load Diffusion pipeline
-    pipe = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16)
+    pipe = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float32)
     # Generate image from text prompt
     prompt = input_text
     generated_image = pipe(prompt).images[0]