Update app.py

app.py

@@ -14,7 +14,9 @@ def rerank(query, docs_text):
     with torch.no_grad():
         scores = model(**inputs).logits.squeeze(-1)
     results = sorted(zip(docs, scores.tolist()), key=lambda x: x[1], reverse=True)
-    return "\n\n".join([f"Score: {score:.4f}\n{doc}" for doc, score in results])
+
+    # Return structured JSON array
+    return [{"score": round(score, 4), "document": doc} for doc, score in results]
 
 # Create API-ready Interface
 iface = gr.Interface(
@@ -23,7 +25,7 @@ iface = gr.Interface(
         gr.Textbox(label="Query", lines=1),
         gr.Textbox(label="Documents (one per line)", lines=10)
     ],
-    outputs="text",
+    outputs="json",
     title="BGE Reranker v2 M3",
     description="Rerank a list of documents based on a search query using BGE v2 M3."
 )