Update app.py

app.py

@@ -52,19 +52,26 @@ project = rf.workspace().project("sku-110k")
 model = project.version(2).model
 
 
-def encode_image(image):
+def resize_image(image, max_size=1500):
     if isinstance(image, np.ndarray):
         image = Image.fromarray(image)
+    if max(image.size) > max_size:
+        ratio = max_size / float(max(image.size))
+        new_size = tuple([int(x * ratio) for x in image.size])
+        image = image.resize(new_size, Image.LANCZOS)
     buffered = io.BytesIO()
-    image.save(buffered, format="PNG")
-    return base64.b64encode(buffered.getvalue()).decode("utf-8")
+    image.save(buffered, format="JPEG")
+    return buffered.getvalue()
+
+def encode_image(image):
+    image_data = resize_image(image)
+    return base64.b64encode(image_data).decode("utf-8")
 
 def roboflow(image, confidence, overlap, stroke_width=1, labels=False):
     '''
         Send the image to Roboflow API for inference.
         Returns JSON and image with bounding boxes drawn on to it.
     '''
-
     json_url = f"https://detect.roboflow.com/sku-110k/2?api_key=gHiUgOSq9GqTnRy5mErk&confidence={confidence}&overlap={overlap}&format=json"
     image_url = f"https://detect.roboflow.com/sku-110k/2?api_key=gHiUgOSq9GqTnRy5mErk&confidence={confidence}&overlap={overlap}&format=image&labels={str(labels).lower()}&stroke={stroke_width}"
     
@@ -93,6 +100,7 @@ def roboflow(image, confidence, overlap, stroke_width=1, labels=False):
     }
 
 
+    
 # Image Splitting and Merging Functionality
 def split_image(image, tile_size=640, overlap=160):
     img_width, img_height = image.size