Image reshape

app.py

@@ -53,7 +53,7 @@ def predict(img_input):
     transforms = [T.ToTensor()]
     # Resize image if needed
     if img.size[0] > MAX_IM_SIZE or img.size[1] > MAX_IM_SIZE:
-        transforms.append(T.Resize(max_size=MAX_IM_SIZE))
+        transforms.append(T.Resize(MAX_IM_SIZE))
     transforms.append(NORMALIZE)
     t = T.Compose(transforms)