Update app.py

app.py

@@ -24,7 +24,7 @@ transform_image = transforms.Compose([
 @spaces.GPU
 def process(image):
     image_size = image.size
-    input_images = transform_image(image).unsqueeze(0).to("cuda")
+    input_images = transform_image(image).unsqueeze(0).to("cpu")
     with torch.no_grad():
         preds = birefnet(input_images)[-1].sigmoid().cpu()
     pred = preds[0].squeeze()