Update src/streamlit_app.py

src/streamlit_app.py

@@ -16,8 +16,12 @@ import requests
 from cloudhands import CloudHandsPayment
 from database_center import db_transaction
 device='cuda' if torch.cuda.is_available() else 'cpu'
-os.environ["HF_HOME"] = "/tmp/huggingface"
-os.environ["TRANSFORMERS_CACHE"] = "/tmp/huggingface"
+import os
+
+# Explicitly override cache paths (matches Dockerfile ENV)
+os.environ["HF_HOME"] = "/app/hf_cache"
+os.environ["TRANSFORMERS_CACHE"] = "/app/hf_cache"
+
 global chat_messages
 chat_messages=[]
 outputs=[]
@@ -143,7 +147,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").to(device)
+    st.session_state.encoder = SentenceTransformer("all-MiniLM-L6-v2",cache_folder="/app/hf_cache").to(device)
 if 'file_text' not in st.session_state:
     st.session_state.file_text = ""
 if "chPay" not in st.session_state: