Initial Commit

app.py

@@ -5,8 +5,11 @@ from diffusers import StableDiffusionPipeline
 model_name = "runwayml/stable-diffusion-v1-5"
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
-pipe = StableDiffusionPipeline.from_pretrained(model_name, torch_dtype=torch.float16)
-pipe = pipe.to(device)
+if device == "cuda":
+    pipe = StableDiffusionPipeline.from_pretrained(model_name, torch_dtype=torch.float16)
+    pipe = pipe.to(device)
+else:
+    pipe = StableDiffusionPipeline.from_pretrained(model_name).to(device)
 
 def generate_image(prompt):
     with torch.no_grad():