Update app.py

app.py

@@ -85,11 +85,14 @@ def readb64(b64):
     img = Image.open(BytesIO(base64.b64decode(b64)))
     return img
     
-# convert from PIL to base64
+# convert from CV2 image to base64 PNG
 def writeb64(image):
-    buffered = BytesIO()
-    image.save(buffered, format="PNG")
-    b64image = base64.b64encode(buffered.getvalue())
+    # this version is for PIL
+    #buffered = BytesIO()
+    #image.save(buffered, format="PNG")
+    #b64image = base64.b64encode(buffered.getvalue())
+    retval, buffer = cv2.imencode('.png', image)
+    b64image = base64.b64encode(buffer)
     b64image_str = b64image.decode("utf-8")
     return b64image_str
     
@@ -543,7 +546,7 @@ def run_anything_task(secret_token, input_image_b64, text_prompt, box_threshold,
     )
     if boxes_filt.size(0) == 0:
         logger.info(f'run_anything_task_[{file_temp}]_{task_type}_[{text_prompt}]_1_[No objects detected, please try others.]_')
-        return [], gr.Gallery.update(label='No objects detected, please try others.😂😂😂😂')
+        return []
     boxes_filt_ori = copy.deepcopy(boxes_filt)