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():
|