Update app.py

app.py

@@ -59,7 +59,7 @@ def process(image: Image.Image) -> Image.Image:
         PIL.Image: The image with the background removed, using the segmentation mask as transparency.
     """
     image_size = image.size
-    input_images = transform_image(image).unsqueeze(0).to("cuda")
+    input_images = transform_image(image).unsqueeze(0).to("cpu")
     # Prediction
     with torch.no_grad():
         preds = birefnet(input_images)[-1].sigmoid().cpu()