Update functions.py

functions.py

@@ -83,7 +83,7 @@ def embed_text(query,corpus,embedding_model):
 
     # ##### Re-Ranking #####
     # Now, score all retrieved passages with the cross_encoder
-    cross_inp = [[search_input, passages[hit['corpus_id']]] for hit in hits]
+    cross_inp = [[search_input, corpus[hit['corpus_id']]] for hit in hits]
 
     if embedding_model == 'hkunlp/instructor-base':
         result = []