Update app.py

app.py

@@ -400,6 +400,14 @@ def get_objects(image, threshold=0.8):
 
     return class_counts, output_img, crops, crop_classes
 
+def encode_images_to_base64(cropped_list):
+    base64_images = []
+    for image in cropped_list:
+        with io.BytesIO() as buffer:
+            image.convert('RGB').save(buffer, format='JPEG')
+            base64_image = base64.b64encode(buffer.getvalue()).decode('utf-8')
+            base64_images.append(base64_image)
+    return base64_images
 
 
 def get_cropped_images(images,category):
@@ -431,7 +439,7 @@ def get_cropped_images(images,category):
             cropped_list.append(image)
 
 
-    base64_images = [base64.b64encode(BytesIO(image.convert('RGB').save(format='JPEG').getvalue())).decode('utf-8') for image in cropped_list]
+    base64_images = encode_images_to_base64(cropped_list)
 
     return base64_images