Update app.py

app.py

@@ -27,7 +27,7 @@ def extract_object(image, t1, t2):
 
     # image = Image.open(imagepath)
     image1 = image.copy()
-    input_images = transform_image(image1).unsqueeze(0).to('cuda').half()
+    input_images = transform_image(image1).unsqueeze(0).to(device).half()
 
     # Prediction
     with torch.no_grad():