Update app

main.py

@@ -92,7 +92,6 @@ def predict(image_input):
 
     # For each mask, get the bounding box.
     for mask in mask_images:
-        return mask.astype(np.uint8)
         bbox_list.extend(get_bbox(mask.astype(np.uint8)))
 
     # Return the bounding boxes.
@@ -103,7 +102,7 @@ def run():
     demo = gr.Interface(
         fn=predict,
         inputs=gr.components.Image(),
-        outputs=gr.components.Image(),
+        outputs=gr.JSON(),
     )
 
     demo.launch(server_name="0.0.0.0", server_port=7860)