Update app.py

app.py

@@ -11,7 +11,7 @@ def generate_and_display_image(prompt_text):
     device = "cuda" if torch.cuda.is_available() else "cpu"
     
     # Create the diffusion pipeline
-    pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
+    pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float32)  # Changed torch.float16 to torch.float32
     pipe = pipe.to(device)
 
     try: