Hugging Face's logo

Spaces:
tanbushi
/
reranker
Sleeping

App Files Community
geqintan commited on
Commit
d794624
·
1 Parent(s): 877dfe6

update

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -55,7 +55,7 @@ async def rerank(
55
  scores = model.predict(pairs)
56
 
57
  results = [
58
- {"index": idx, "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)
 
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)