Update app.py

app.py

@@ -17,13 +17,13 @@ def predict(image):
     width, height = image.size
     ratio = width / height
     if(width > height):
-      height = int(256 / ratio)
-      width = 256
+      new_height = int(256 / ratio)
+      new_width = 256
     else:
-      width = int(256 * ratio)
-      height = 256
+      new_width = int(256 * ratio)
+      new_height = 256
     
-    resized_image = TF.resize(image, (height, width))
+    resized_image = TF.resize(image, (new_height, new_width))
 
     padded_image = Image.new("RGB", (256, 256))
     padded_image.paste(resized_image, (0, 0))