Update app.py

app.py

@@ -100,11 +100,13 @@ image_width = 128
 image_height = 32
 
 def distortion_free_resize(image, img_size):
-  w, h = img_size
-  image = tf.image.resize(image, size=(h, w), preserve_aspect_ratio=True)
-  image = tf.transpose(image, perm=[1,0,2])
-  image = tf.image.flip_left_right(image)
-  return image
+    w, h = img_size
+    image = tf.image.resize(image, size=(h, w), preserve_aspect_ratio=True)
+    image = tf.image.resize_with_pad(image, target_height=h, target_width=w)
+    image = tf.transpose(image, perm=[1, 0, 2])
+    image = tf.image.flip_left_right(image)
+    return image
+
 
 def decode_batch_predictions(input_image, img_size=(image_width, image_height)):
     img_byte_array = io.BytesIO()