Create resize function

app.py

@@ -5,9 +5,26 @@ import tempfile
 import os
 import yaml
 
+def resize_image(image, base_height):
+
+    # Calculate aspect ratio
+    w_percent = base_height / float(image.size[1])
+    w_size = int(float(image.size[0]) * float(w_percent))
+
+    # Resize the image
+    return image.resize((w_size, base_height), Image.ANTIALIAS)
+
 def predict(input_image: Image.Image):
 
     try:
+
+        # Try to resize the image to a fixed height of 128 pixels
+        try:
+            input_image = resize_image(input_image, 128)
+        except Exception as e:
+            print(f"Image resizing failed: {e}")
+            return f"Image resizing failed: {e}"
+
         # Used as a context manager. Takes care of cleaning up the directory.
         # Even if an error is raised within the with block, the directory is removed.
         # No finally block needed