Update app.py

app.py

@@ -13,10 +13,17 @@ import matplotlib.pyplot as plt
 # Load the Stable Diffusion model for text-to-image generation and inpainting
 device = "cuda" if torch.cuda.is_available() else "cpu"
 pipeline = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4").to(device)
-pipeline_inpaint = StableDiffusionInpaintPipeline.from_pretrained(
-    "runwayml/stable-diffusion-inpainting", 
-    torch_dtype=torch.float16
-).to(device)
+
+# Adjusted to handle device compatibility
+if device == "cuda":
+    pipeline_inpaint = StableDiffusionInpaintPipeline.from_pretrained(
+        "runwayml/stable-diffusion-inpainting", 
+        torch_dtype=torch.float16
+    ).to(device)
+else:
+    pipeline_inpaint = StableDiffusionInpaintPipeline.from_pretrained(
+        "runwayml/stable-diffusion-inpainting"
+    ).to(device)
 
 DEFAULT_OBJ_FILE = "female.obj"
 DEFAULT_GLB_FILE = "vroid_girl1.glb"