Update app.py

app.py

@@ -127,13 +127,14 @@ def distortion_free_resize(image, img_size):
           [0, 0],
       ],)
   image = tf.transpose(image, perm=[1,0,2])
+  image = tf.image.resize_with_crop_or_pad(image, target_height=128, target_width=32)
   image = tf.image.flip_left_right(image)
   return image
 
 def decode_batch_predictions(input_image):
     img_size=(128, 32)
     img_byte_array = io.BytesIO()
-    input_image.save(img_byte_array, format='JPEG')  # Use 'JPG'  # Change the format as needed
+    input_image.save(img_byte_array, format='JPEG')  # Change the format as needed
     input_image = img_byte_array.getvalue()
     input_image = tf.io.decode_image(input_image, channels=1,  dtype=tf.dtypes.uint8) 
     input_image = distortion_free_resize(input_image, img_size)