Update chat_a.py

chat_a.py

@@ -12,9 +12,10 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
 from css import log_and_render
 
 # ──────────────────────────────── 캐시/데이터셋 설정 ────────────────────────────────
-APP_DIR = pathlib.Path(__file__).parent.resolve()
+HOME = pathlib.Path.home()
 
-CACHE_DIR = os.getenv("TRANSFORMERS_CACHE") or str(APP_DIR / ".cache" / "hf-cache")
+# ✅ ENV가 있으면 따르고, 없으면 홈 밑 .cache/hf-cache 사용
+CACHE_DIR = os.getenv("TRANSFORMERS_CACHE") or str(HOME / ".cache" / "hf-cache")
 os.makedirs(CACHE_DIR, exist_ok=True)
 
 HF_DATASET_REPO = os.getenv("HF_DATASET_REPO", "emisdfde/moai-travel-data")
@@ -109,7 +110,9 @@ def load_tokenizer():
 @st.cache_resource(show_spinner=False)
 def load_sentiment_model():
     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 = AutoModelForSequenceClassification.from_pretrained(
+        "hun3359/klue-bert-base-sentiment", cache_dir=CACHE_DIR
+    )
     model.eval()
     return tok, model