Update app.py

app.py

@@ -35,7 +35,7 @@ def remove_obj(image, uploaded_mask, seed):
     image_pil, sketched_mask = image["image"], image["mask"]
     mask = dilate_mask(combine_masks(uploaded_mask, sketched_mask))
     seed = int(seed)
-    latents = torch.randn((1, 4, 64, 64), generator=torch.Generator().manual_seed(seed)).to("cuda")
+    latents = torch.randn((1, 4, 64, 64), generator=torch.Generator().manual_seed(seed)).to("cpu")
     final_image = clipaway.generate(
         prompt=[""], scale=1, seed=seed,
         pil_image=[image_pil], alpha=[mask], strength=1, latents=latents