Update app.py

app.py

@@ -29,7 +29,7 @@ def fn(image):
     image_size = im.size
     origin = im.copy()
     image = load_img(im)
-    input_images = transform_image(image).unsqueeze(0).to(birefnet)
+    input_images = transform_image(image).unsqueeze(0).to(device)
     # Prediction
     with torch.no_grad():
         preds = birefnet(input_images)[-1].sigmoid().cpu()