Update app.py

app.py

@@ -14,8 +14,8 @@ model = model.cpu()
 
 def transform_image(image):
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-    my_transforms = transformss.Compose([transformss.ToTensor(),
-                                        transformss.Normalize(
+    my_transforms = transforms.Compose([transforms.ToTensor(),
+                                        transforms.Normalize(
                                             [0.485, 0.456, 0.406],
                                             [0.229, 0.224, 0.225])])
     image_aux = image
@@ -23,7 +23,7 @@ def transform_image(image):
 
 def predict(img):
     img = PIL.Image.fromarray(img, "RGB")
-    image = transformss.Resize((480,640))(img)
+    image = transforms.Resize((480,640))(img)
     tensor = transform_image(image=image) 
     
     learner.to(device)