gr json

app.py

@@ -1,17 +1,22 @@
 import gradio as gr
-from sentence_transformers.cross_encoder import CrossEncoder
+from sentence_transformers import CrossEncoder
 
 ce = CrossEncoder("cross-encoder/ms-marco-MiniLM-L6-v2")
 
-def rerank(query, docs_list):
-    pairs = [[query, doc] for doc in docs_list]
+def rerank(query, docs):
+    # docs уже list[dict], а не строка
+    texts = [d.get("A", str(d)) for d in docs]
+    pairs = [[query, txt] for txt in texts]
     scores = ce.predict(pairs)
-    rows = [[doc, float(score)] for doc, score in zip(docs_list, scores)]
+    rows = [[txt, float(score)] for txt, score in zip(texts, scores)]
     return rows
 
 iface = gr.Interface(
     fn=rerank,
-    inputs=[gr.Textbox(label="Query"), gr.Textbox(label="Docs (JSON‑list)")],
+    inputs=[
+        gr.Textbox(label="Query"),
+        gr.JSON(label="Docs (JSON array of objects)")
+    ],
     outputs=gr.Dataframe(type="array", headers=["doc", "score"]),
     api_name="rerank"
 )