update
Browse files
app.py
CHANGED
|
@@ -55,9 +55,13 @@ async def rerank(
|
|
| 55 |
scores = model.predict(pairs)
|
| 56 |
|
| 57 |
results = [
|
| 58 |
-
|
| 59 |
-
|
| 60 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 61 |
sorted_results = sorted(results, key=lambda x: x["score"], reverse=True)
|
| 62 |
|
| 63 |
if request.top_k is not None:
|
|
|
|
| 55 |
scores = model.predict(pairs)
|
| 56 |
|
| 57 |
results = [
|
| 58 |
+
{
|
| 59 |
+
"index": idx,
|
| 60 |
+
"relevance_score": float(score), # 同步修改键名
|
| 61 |
+
"document": doc
|
| 62 |
+
}
|
| 63 |
+
for idx, (doc, score) in enumerate(zip(request.documents, scores))
|
| 64 |
+
]
|
| 65 |
sorted_results = sorted(results, key=lambda x: x["score"], reverse=True)
|
| 66 |
|
| 67 |
if request.top_k is not None:
|