Update app.py

app.py

@@ -30,11 +30,9 @@ def perform(image, task: str="detect"):
         confs = float(result.boxes.conf)  # confidence score of each box
         outputs.append({"xywh": xywh, "conf": confs, "name": name})
     
-    # Add base64 image to outputs
-    outputs.append({"image_base64": f"data:image/jpeg;base64,{img_base64}"})
     
     # return the PIL image (for display) and the outputs with base64
-    return prediction_image, outputs
+    return prediction_image, outputs, {"image_base64": f"data:image/jpeg;base64,{img_base64}"}
 
 def load_model(model_name: str = "yolo11n.pt"):
     """Model loading"""
@@ -52,7 +50,7 @@ model = load_model()
 demo = gr.Interface(
     fn=perform,
     inputs=[gr.Image(type="pil")],  # PIL Image format
-    outputs=[gr.Image(type="pil"), gr.JSON()],  # Back to PIL - Gradio handles public URL automatically
+    outputs=[gr.Image(type="pil"), gr.JSON(), gr.JSON()],
     title="Object Detection using YOLO",
     description="Upload an image and get image with bbox in it",
     examples=None  # You can add example images here if needed