Update chat_a.py

chat_a.py

@@ -2,7 +2,7 @@
 # coding: utf-8
 
 # In[10]:
-import os, io
+import os, io, pathlib
 from huggingface_hub import hf_hub_download
 import pandas as pd
 import torch
@@ -14,6 +14,9 @@ from datetime import datetime
 import random
 import re
 
+CACHE_DIR = os.getenv("TRANSFORMERS_CACHE", "/tmp/hf-cache")
+os.makedirs(CACHE_DIR, exist_ok=True)
+
 HF_DATASET_REPO = os.getenv("HF_DATASET_REPO", "emisdfde/moai-travel-data")  # ← 본인 리포
 HF_DATASET_REV  = os.getenv("HF_DATASET_REV", "main")
 
@@ -60,6 +63,13 @@ from css import log_and_render
 import streamlit as st, pandas as pd, json, requests
 # -------------------- 모델 및 데이터 로딩 --------------------
 # 모델 로딩 부분을 함수로 만들고 데코레이터 추가
+@st.cache_resource(show_spinner=False)
+def load_sentiment_model():
+    repo = "hun3359/klue-bert-base-sentiment"
+    tok = AutoTokenizer.from_pretrained(repo, cache_dir=CACHE_DIR)
+    model = AutoModelForSequenceClassification.from_pretrained(repo, cache_dir=CACHE_DIR)
+    return tok, model
+    
 @st.cache_resource
 def load_sbert_model():
     print("SBERT 모델 로딩 중... (이 메시지는 한 번만 보여야 합니다)")