Update app.py

app.py

@@ -19,15 +19,10 @@ from langchain.memory import ConversationBufferMemory
 from langchain_core.runnables.history import RunnableWithMessageHistory
 
 
-@st.cache_data
+@st.cache_data(persist=False)
 def get_df():
     return pickle.load(open("ebesha_ticket_df.pkl", "rb"))
 
-@st.cache_data
-def get_unique_values():
-    exec(custom_unique_df_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).invoke({"df_example":format_df(st.session_state.df.head(4))}))
-    return response
-
 
 @st.cache_resource
 def get_llm_chain():
@@ -44,8 +39,8 @@ if 'df' not in st.session_state:
 
 
 if 'unique_values' not in st.session_state:
-    
-    st.session_state.unique_values = get_unique_values()
+    exec(custom_unique_df_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).invoke({"df_example":format_df(st.session_state.df.head(4))}))
+    st.session_state.unique_values = response