feat: add outputs

.gitignore

@@ -1,2 +1,4 @@
 .DS_Store
-yolov5s.pt
+yolov5s.pt
+__pycache__
+gradio_queue.db

app.py

@@ -1,8 +1,8 @@
 import gradio as gr
 import torch
 from PIL import Image
-
-import gdown
+import json
+# import gdown
 
 '''
 # a file
@@ -29,11 +29,16 @@ def yolo(im, size=1024):
 
     results = model(im)  # inference
     results.render()  # updates results.imgs with boxes and labels
-    return Image.fromarray(results.imgs[0])
+
+    df = results.pandas().xyxy[0].to_json(orient="records")
+    res = json.loads(df)
+
+    return [Image.fromarray(results.imgs[0]), res]
 
 
 inputs = gr.inputs.Image(type='pil', label="Original Image")
-outputs = gr.outputs.Image(type="pil", label="Output Image")
+outputs = [gr.outputs.Image(type="pil", label="Output Image"),
+    gr.outputs.JSON(label="Output JSON")]
 
 title = "YOLOv5 Character"
 description = "YOLOv5 Character Gradio demo for object detection. Upload an image or click an example image to use."