Update app.py

app.py

@@ -75,8 +75,8 @@ def transform_image(image):
 
 def predict(img):
     img = PIL.Image.fromarray(img, "RGB")
-    image = transforms.Scale((480,640))(img)
-    tensor = transform_image(image=image) 
+    #image = transforms.Resize((480,640))(img)
+    tensor = transform_image(image=img) 
 
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     learner.to(device)