Update app.py

app.py

@@ -15,8 +15,8 @@ import torch
 
 # Load Stable Diffusion Model
 def load_stable_diffusion_model():
-    device = "cuda" if torch.cuda.is_available() else "cpu"
-    return StableDiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-2-1", torch_dtype=torch.float32).to(device)
+    pipe = StableDiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-2-1", torch_dtype=torch.float32)
+    pipe = pipe.to("cpu") 
 
 sd_model = load_stable_diffusion_model()