Update app.py

app.py

@@ -26,8 +26,9 @@ def get_df():
 
 @st.cache_resource
 def get_llm_chain():
-    dataframe_chain = custom_dataframe_chain(llm=CustomLLM(repo_id="mistralai/Mixtral-8x7B-Instruct-v0.1", model_type='text-generation', api_token=API_TOKEN, stop=["\n<|","<|"]), df=st.session_state.df, unique_values=st.session_state.unique_values)
-    memory_chain = custom_chain_with_history(llm=CustomLLM(repo_id="mistralai/Mixtral-8x7B-Instruct-v0.1", model_type='text-generation', api_token=API_TOKEN, stop=["\n<|","<|"]), memory=st.session_state.memory)
+    llm = CustomLLM(repo_id="mistralai/Mixtral-8x7B-Instruct-v0.1", model_type='text-generation', api_token=API_TOKEN, stop=["\n<|","<|"])
+    dataframe_chain = custom_dataframe_chain(llm=llm, df=st.session_state.df, unique_values=st.session_state.unique_values)
+    memory_chain = custom_chain_with_history(llm=llm, memory=st.session_state.memory)
     return custom_unique_df_chain(llm=llm, dataframe_chain=dataframe_chain, memory_chain=memory_chain)
 
 if 'memory' not in st.session_state: