Update app.py

app.py

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