Update app.py

app.py

@@ -1,10 +1,9 @@
 import gradio as gr
 from diffusers import StableDiffusionPipeline
-import torch
 
 # Load the model and set up the pipeline
 model_id = "prompthero/openjourney"
-pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
+pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float32)
 
 def generate_image(prompt):
     image = pipe(prompt).images[0]