Update pages/3_chat.py

pages/3_chat.py

@@ -41,7 +41,7 @@ PROMPT = PromptTemplate(
 # chain_type_kwargs = {"prompt": PROMPT}
 
 @st.cache_resource
-def load_conversational_qa_memory_retriever(llm_model, vector_database):
+def load_conversational_qa_memory_retriever():
 
     question_generator = LLMChain(llm=llm_model, prompt=CONDENSE_QUESTION_PROMPT)
     doc_chain = load_qa_chain(llm_model, chain_type="stuff", prompt = PROMPT)
@@ -86,10 +86,9 @@ if "chat_history" not in st.session_state: #this one is to pass previous message
     st.session_state.chat_history = []
 
 
-
-
-conversational_qa_memory_retriever, question_generator = load_conversational_qa_memory_retriever(llm_model =  st.session_state['model'],
-                                                                                                 vector_database =  st.session_state['faiss_db'])
+llm_model =  st.session_state['model'],
+vector_database =  st.session_state['faiss_db']
+conversational_qa_memory_retriever, question_generator = load_conversational_qa_memory_retriever()