bhulston commited on
Commit
26ed99e
1 Parent(s): 9f17ce8

Update reranker.py

Browse files
Files changed (1) hide show
  1. reranker.py +1 -1
reranker.py CHANGED
@@ -14,5 +14,5 @@ def reranker(query, matches):
14
 
15
  scores = model.predict(pairs)
16
  top_indices = np.argsort(scores)[::-5]
17
- top_results = ["Class: " + docs[index]["metadata"]["text"] for index in top_indices]
18
  return top_results
 
14
 
15
  scores = model.predict(pairs)
16
  top_indices = np.argsort(scores)[::-5]
17
+ top_results = ["Class: " + matches[index]["metadata"]["text"] for index in top_indices]
18
  return top_results