bhulston commited on
Commit
56eebbb
1 Parent(s): 9f4bfb9

Update reranker.py

Browse files
Files changed (1) hide show
  1. reranker.py +1 -1
reranker.py CHANGED
@@ -7,7 +7,7 @@ def reranker(query, matches):
7
 
8
  pairs = []
9
  for match in matches:
10
- pairs.append((query, match["metadata"]["text"]))
11
 
12
  model = CrossEncoder('cross-encoder/ms-marco-TinyBERT-L-2-v2', max_length = 512)
13
 
 
7
 
8
  pairs = []
9
  for match in matches:
10
+ pairs.append([query, match["metadata"]["text"]])
11
 
12
  model = CrossEncoder('cross-encoder/ms-marco-TinyBERT-L-2-v2', max_length = 512)
13