Spaces:
Sleeping
Sleeping
wenbemi
commited on
Update chat_a.py
Browse files
chat_a.py
CHANGED
|
@@ -109,12 +109,12 @@ def load_tokenizer():
|
|
| 109 |
|
| 110 |
@st.cache_resource(show_spinner=False)
|
| 111 |
def load_sentiment_model():
|
| 112 |
-
|
| 113 |
model = AutoModelForSequenceClassification.from_pretrained(
|
| 114 |
"hun3359/klue-bert-base-sentiment", cache_dir=CACHE_DIR
|
| 115 |
)
|
| 116 |
model.eval()
|
| 117 |
-
return
|
| 118 |
|
| 119 |
@st.cache_resource(show_spinner=False)
|
| 120 |
def load_sbert_model():
|
|
|
|
| 109 |
|
| 110 |
@st.cache_resource(show_spinner=False)
|
| 111 |
def load_sentiment_model():
|
| 112 |
+
# ํ ํฌ๋์ด์ ๋ load_tokenizer ํจ์๊ฐ ๋ด๋นํ๋ฏ๋ก ์ฌ๊ธฐ์๋ ๋ชจ๋ธ๋ง ๋ก๋
|
| 113 |
model = AutoModelForSequenceClassification.from_pretrained(
|
| 114 |
"hun3359/klue-bert-base-sentiment", cache_dir=CACHE_DIR
|
| 115 |
)
|
| 116 |
model.eval()
|
| 117 |
+
return model # ๋ชจ๋ธ๋ง ๋ฐํํ๋๋ก ์์ !
|
| 118 |
|
| 119 |
@st.cache_resource(show_spinner=False)
|
| 120 |
def load_sbert_model():
|