nickmuchi commited on
Commit
a4a68cf
1 Parent(s): cd6c5de

Update functions.py

Browse files
Files changed (1) hide show
  1. functions.py +1 -1
functions.py CHANGED
@@ -83,7 +83,7 @@ def embed_text(query,corpus,embedding_model):
83
 
84
  # ##### Re-Ranking #####
85
  # Now, score all retrieved passages with the cross_encoder
86
- cross_inp = [[search_input, passages[hit['corpus_id']]] for hit in hits]
87
 
88
  if embedding_model == 'hkunlp/instructor-base':
89
  result = []
 
83
 
84
  # ##### Re-Ranking #####
85
  # Now, score all retrieved passages with the cross_encoder
86
+ cross_inp = [[search_input, corpus[hit['corpus_id']]] for hit in hits]
87
 
88
  if embedding_model == 'hkunlp/instructor-base':
89
  result = []