Update chat_a.py

chat_a.py

@@ -12,7 +12,9 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
 from css import log_and_render
 
 # ──────────────────────────────── 캐시/데이터셋 설정 ────────────────────────────────
-CACHE_DIR = os.getenv("TRANSFORMERS_CACHE", "/data/hf-cache")
+APP_DIR = pathlib.Path(__file__).parent.resolve()
+
+CACHE_DIR = os.getenv("TRANSFORMERS_CACHE") or str(APP_DIR / ".cache" / "hf-cache")
 os.makedirs(CACHE_DIR, exist_ok=True)
 
 HF_DATASET_REPO = os.getenv("HF_DATASET_REPO", "emisdfde/moai-travel-data")
@@ -106,18 +108,14 @@ def load_tokenizer():
 
 @st.cache_resource(show_spinner=False)
 def load_sentiment_model():
-    model = AutoModelForSequenceClassification.from_pretrained(
-        "hun3359/klue-bert-base-sentiment",
-        cache_dir=CACHE_DIR
-    )
+    tok = AutoTokenizer.from_pretrained("hun3359/klue-bert-base-sentiment", cache_dir=CACHE_DIR)
+    model = AutoModelForSequenceClassification.from_pretrained("hun3359/klue-bert-base-sentiment", cache_dir=CACHE_DIR)
     model.eval()
-    return model
+    return tok, model
 
 @st.cache_resource(show_spinner=False)
 def load_sbert_model():
-    # SentenceTransformer 쪽도 캐시 폴더 명시
-    return SentenceTransformer("jhgan/ko-sroberta-multitask",
-                               cache_folder=CACHE_DIR)
+    return SentenceTransformer("jhgan/ko-sroberta-multitask", cache_folder=CACHE_DIR)
     
 def detect_location_filter(text, intent_score=None):
     def in_text_exact(word):