Update ui/ui_core.py

ui/ui_core.py

@@ -7,10 +7,17 @@ import gradio as gr
 from typing import List
 from concurrent.futures import ThreadPoolExecutor, as_completed
 import hashlib
+import shutil
 
 # ✅ Fix: Add src to Python path
 sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "..", "src")))
 
+# ✅ Persist model cache to Hugging Face Space's /data directory
+model_cache_dir = "/data/txagent_models"
+os.makedirs(model_cache_dir, exist_ok=True)
+os.environ["TRANSFORMERS_CACHE"] = model_cache_dir
+os.environ["HF_HOME"] = model_cache_dir
+
 from txagent.txagent import TxAgent
 
 def sanitize_utf8(text: str) -> str: