Update app.py

app.py

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