Update app.py

app.py

@@ -12,10 +12,8 @@ from huggingface_hub import login
 
 
 
-@st.cache_data
+@st.cache_resource
 def load_data():
-   
-    
     retriever = BM25HF.load_from_hub(
     "tien314/bm25s-version2", load_corpus=True, mmap=True)
     return retriever
@@ -43,7 +41,7 @@ def load_model():
     return chain
 
 def process_input(sentence):
-    docs, _ = retriever.retrieve(bm25s.tokenize(sentence), k=15)
+    docs, _ = st.session_state.retriever.retrieve(bm25s.tokenize(sentence), k=15)
     documents =[]
     for doc in docs[0]:
         documents.append(Document(doc['text'])) 
@@ -65,6 +63,6 @@ sentence = st.text_input("please enter description:")
 
 if sentence !='':
     documents = process_input(sentence)
-    hscode = chain.invoke({'context': documents,'description':sentence})
+    hscode = st.session_state.chain.invoke({'context': documents,'description':sentence})
     st.write("answer:",hscode)