update

model.py

@@ -32,10 +32,15 @@ def segment_image(image):
 
 # Image inpainting
 # get Stable Diffusion model for image inpainting
-pipe = StableDiffusionInpaintPipeline.from_pretrained(
-    "runwayml/stable-diffusion-inpainting",
-    torch_dtype=torch.float16,
-).to(device)
+if device == 'cuda':
+    pipe = StableDiffusionInpaintPipeline.from_pretrained(
+        "runwayml/stable-diffusion-inpainting",
+        torch_dtype=torch.float16,
+    ).to(device)
+else:
+    pipe = StableDiffusionInpaintPipeline.from_pretrained(
+        "runwayml/stable-diffusion-inpainting"
+    ).to(device)
 
 
 def inpaint(image, mask, W, H, prompt="", seed=0, guidance_scale=17.5, num_samples=3):