feat: update gradio

README.md

@@ -4,7 +4,7 @@ emoji: 🌍
 colorFrom: yellow
 colorTo: purple
 sdk: gradio
-sdk_version: 4.8.0
+sdk_version: 4.44.1
 app_file: app.py
 pinned: false
 ---

app.py

@@ -10,12 +10,8 @@ def yolo(im):
     results = model(im)  # inference
 
     df = results.pandas().xyxy[0].to_json(orient="records")
-    res = json.loads(df)
 
-    im_with_boxes = results.render()[0]  # results.render() returns a list of images
-    
-    # Convert the numpy array back to an image
-    output_image = Image.fromarray(im_with_boxes)
+    res = json.loads(df)
 
     draw = ImageDraw.Draw(im)
 
@@ -27,7 +23,6 @@ def yolo(im):
         draw.rectangle([xmin, ymin, xmax, ymax], outline="red", width=3)
 
     return [
-        output_image,
         res,
         im,
     ]
@@ -35,7 +30,6 @@ def yolo(im):
 
 inputs = gr.Image(type='pil', label="Original Image")
 outputs = [
-    gr.Image(type="pil", label="Output Image"), 
     gr.JSON(label="Output JSON"),
     gr.Image(type='pil', label="Output Image with Boxes"),
 ]

requirements.txt

@@ -1 +1,2 @@
-yolov5
+yolov5
+gradio