Update app.py

app.py

@@ -5,12 +5,12 @@ import gradio as gr
 
 model_id = "stabilityai/stable-diffusion-2-1"
 pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
-pipe = pipe.to("cuda")
+#pipe = pipe.to("cuda")
+pipe = pipe.to("cpu")
 
 
 def generate_image(prompt):
-    with torch.autocast("cuda"):
-        image = pipe(prompt).images[0]  
+    image = pipe(prompt).images[0]  
     return image