Update app.py

app.py

@@ -1,5 +1,6 @@
 import pandas as pd
 import bm25s
+from bm25s.hf import BM25HF
 from langchain_core.prompts import ChatPromptTemplate, HumanMessagePromptTemplate
 from langchain.docstore.document import Document
 import torch
@@ -8,6 +9,8 @@ from langchain.chains import LLMChain
 
 @st.cache_data
 def load_data():
+    retriever = BM25HF.load_from_hub(
+    "tien314/bm25s-version2", load_corpus=True, mmap=True)
     retriever = bm25s.BM25.load("bm25s_very_big_index", mmap=True, load_corpus = True)
     return retriever