Update app.py

app.py

@@ -67,6 +67,7 @@ repo_id = "paascorb/practica3_Segmentation"
 learner = from_pretrained_fastai(repo_id)
 
 def transform_image(image):
+    device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     my_transforms = transformss.Compose([transformss.ToTensor(),
                                         transformss.Normalize(
                                             [0.485, 0.456, 0.406],