Update app.py

app.py

@@ -32,15 +32,19 @@ def predict(base64: str, queries: str, box_threshold: float, text_threshold: flo
         text_threshold=text_threshold,
         target_sizes=[image.size[::-1]]
     )
-    return results
+    return {
+        "scores": [float(s) for s in results[0]["scores"]],
+        "labels": results[0]["labels"],
+        "boxes": [[float(x) for x in box] for row in results[0]["boxes"]]
+    }
 
 demo = gr.Interface(
     fn=predict,
     inputs=[
         gr.Text(label="Image (B64)"),
         gr.Text(label="Queries, in lowercase, separated by full stop", placeholder="a bird. a blue bird."),
-        gr.Number(label="box_threshold"),
-        gr.Number(label="text_threshold")
+        gr.Number(label="box_threshold", placeholder=0.4),
+        gr.Number(label="text_threshold", placeholder=0.3)
     ],
     outputs=gr.JSON(label="Predictions"),
 )