Update app.py

app.py

@@ -12,7 +12,7 @@ def inpaint_with_mask(image: Image.Image, mask: Image.Image, prompt: str = "back
         "diffusers/stable-diffusion-xl-1.0-inpainting-0.1",
         torch_dtype=torch.float16,
         variant="fp16"
-    ).to("cuda")
+    ).to("cpu")
 
     # 🖌️ Inpaint
     result = pipe(prompt=prompt, image=image, mask_image=mask).images[0]
@@ -27,7 +27,7 @@ def inpaint_with_mask(image: Image.Image, mask: Image.Image, prompt: str = "back
         "stabilityai/stable-diffusion-xl-refiner-1.0",
         torch_dtype=torch.float16,
         variant="fp16"
-    ).to("cuda")
+    ).to("cpu")
 
     result = refiner(prompt=prompt, image=result).images[0]