Update app.py

return as base64

app.py

@@ -24,7 +24,18 @@ from segment_anything import build_sam, SamPredictor
 import spaces
 import GroundingDINO.groundingdino.datasets.transforms as T
 from GroundingDINO.groundingdino.util.utils import clean_state_dict, get_phrases_from_posmap
+import base64
+from io import BytesIO
+from PIL import Image
 
+def image_to_base64(image: Image.Image, format="PNG") -> str:
+    """Convert a PIL Image to a base64 string."""
+    buffer = BytesIO()
+    image.save(buffer, format=format)
+    buffer.seek(0)
+    img_bytes = buffer.read()
+    base64_str = base64.b64encode(img_bytes).decode("utf-8")
+    return base64_str
 
 
 # GroundingDINO config and checkpoint
@@ -336,10 +347,8 @@ def run_local(base_image, base_mask, reference_image, ref_mask, seed, base_mask_
     edited_image = crop_back(edited_image, old_tar_image, np.array([H1, W1, H2, W2]), np.array(tar_box_yyxx_crop)) 
     edited_image = Image.fromarray(edited_image)
 
-    if ref_mask_option != "Label to Mask":
-        return [show_diptych_ref_tar, edited_image]
-    else:
-        return [return_ref_mask, show_diptych_ref_tar, edited_image]   
+  
+    return image_to_base64(edited_image)
 
 def update_ui(option):
     if option == "Draw Mask":