Update src/streamlit_app.py

src/streamlit_app.py

@@ -206,6 +206,12 @@ def initialize_pinecone():
     index = pc.Index("rag-v2")
     return index
 
+@st.cache_resource(show_spinner=False)
+def initialize_bm25():
+    with open("bm25_model.pkl", "rb") as f:
+        bm25 = pickle.load(f)
+    retrun bm25
+
 @st.cache_resource(show_spinner=False)
 def initialize_groq():
     client = Groq(api_key=GROQ_API_KEY)
@@ -226,8 +232,7 @@ if "chat_mode" not in st.session_state:
 # ------------------------------- 
 def search_documents(query, filter_dict, top_k):
     """Search for relevant documents using embedding similarity and reranking."""
-    # Encode query
-    bm25 = BM25Encoder()
+
     dense_query = embed_model.encode(query).tolist()
     sparse_query = bm25.encode_queries([query])[0]