Update app.py

app.py

@@ -11,7 +11,7 @@ from fastai.callback import *
 from pathlib import Path
 import random
 import PIL
-import torchvision.transforms as transforms
+import torchvision.transforms as transformss
 from albumentations import *
 
 def get_y_fn (x):
@@ -66,7 +66,7 @@ repo_id = "paascorb/practica3_Segmentation"
 learner = from_pretrained_fastai(repo_id)
 
 def transform_image(image):
-    my_transforms = transforms.Compose([transforms.ToTensor(),
+    my_transforms = transformss.Compose([transforms.ToTensor(),
                                         transforms.Normalize(
                                             [0.485, 0.456, 0.406],
                                             [0.229, 0.224, 0.225])])
@@ -75,7 +75,7 @@ def transform_image(image):
 
 def predict(img):
     img = PIL.Image.fromarray(img, "RGB")
-    #image = transforms.Resize((480,640))(img)
+    image = transformss.Resize((480,640))(img)
     tensor = transform_image(image=img) 
 
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")