Update app.py

app.py

@@ -32,15 +32,11 @@ def predict(image, threshold=0.25, model_id=None):
     # perform inference
     model.conf = threshold
     results = model(image, size=input_size)
-    
-    output_json = []
-    for i in results.pred:
-        output_json.append({"boxes": i[:, :4], "scores": i[:, 4], "categories": i[:, 5]})
 
     numpy_image = results.render()[0]
     output_image = Image.fromarray(numpy_image)
     
-    return (output_image, json.dumps(output_json))
+    return (output_image, results.pandas().xyxy[0].to_json())
 
 
 gr.Interface(