Update app.py

app.py

@@ -38,9 +38,9 @@ def bm25_tokenizer(text):
             tokenized_doc.append(token)
     return tokenized_doc
 
-def retrieval(query, top_k_retriver, docs):
+def retrieval(query, top_k_retriver, docs, bm25_):
 
-    bm25_scores = bm25.get_scores(bm25_tokenizer(query))
+    bm25_scores = bm25_.get_scores(bm25_tokenizer(query))
     top_n = np.argsort(bm25_scores)[::-1][:top_k_retriver]
     bm25_hits = [{'corpus_id': idx, 
                   'score': bm25_scores[idx], 
@@ -70,7 +70,7 @@ def final_qa_pipeline(file, query):
     bm25 = BM25Okapi(tokenized_corpus)
     
     top_k_retriver, top_k_ranker = 10,1
-    lvl1 = retrieval(query, top_k_retriver, docs)
+    lvl1 = retrieval(query, top_k_retriver, docs, bm25)
 
     if len(lvl1) > 0:
         fnl_rank = qa_ranker(query, [l["docs"] for l in lvl1], top_k_ranker)