Update app.py

app.py

@@ -121,14 +121,14 @@ def get_y_fn (x):
     return Path(str(x).replace("Images","Labels").replace("color","gt").replace(".jpg",".png"))
     
 learn = from_pretrained_fastai("islasher/segm-grapes")
-
+#device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 def transform_image(image):
     my_transforms = transforms.Compose([transforms.ToTensor(),
                                         transforms.Normalize(
                                             [0.485, 0.456, 0.406],
                                             [0.229, 0.224, 0.225])])
     image_aux = image
-    return my_transforms(image_aux).unsqueeze(0).to(device)
+    return my_transforms(image_aux).unsqueeze(0)#.to(device)