Update app.py

app.py

@@ -25,7 +25,7 @@ transform_image = transforms.Compose([
 
 def process(image: Image.Image) -> Image.Image:
     image_size = image.size
-    input_tensor = transform_image(image).unsqueeze(0).to("cuda")
+    input_tensor = transform_image(image).unsqueeze(0).to(device)
 
     with torch.no_grad():
         preds = birefnet(input_tensor)[-1].sigmoid().cpu()