Update src/streamlit_app.py

src/streamlit_app.py

@@ -22,6 +22,12 @@ import os
 os.environ["HF_HOME"] = "/app/hf_cache"
 os.environ["TRANSFORMERS_CACHE"] = "/app/hf_cache"
 
+encoder = SentenceTransformer(
+    "sentence-transformers/all-MiniLM-L6-v2",
+    cache_folder="/app/hf_cache"
+).to(device)
+
+
 global chat_messages
 chat_messages=[]
 outputs=[]
@@ -147,7 +153,7 @@ if 'doc_flag' not in st.session_state:
 if 'flag' not in st.session_state:
     st.session_state.flag = False
 if 'encoder' not in st.session_state:
-    st.session_state.encoder = SentenceTransformer("all-MiniLM-L6-v2",cache_folder="/app/hf_cache").to(device)
+    st.session_state.encoder = encoder
 if 'file_text' not in st.session_state:
     st.session_state.file_text = ""
 if "chPay" not in st.session_state: