geqintan commited on
Commit
95d656f
·
1 Parent(s): d794624
Files changed (1) hide show
  1. app.py +7 -3
app.py CHANGED
@@ -55,9 +55,13 @@ async def rerank(
55
  scores = model.predict(pairs)
56
 
57
  results = [
58
- {"index": idx, "relevance_score": float(score), "document": doc}
59
- for idx, (doc, score) in enumerate(zip(request.documents, scores))
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: