wenbemi commited on
Commit
4c030ae
ยท
verified ยท
1 Parent(s): 40e0b51

Update chat_a.py

Browse files
Files changed (1) hide show
  1. chat_a.py +2 -2
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
- tok = AutoTokenizer.from_pretrained("hun3359/klue-bert-base-sentiment", cache_dir=CACHE_DIR)
113
  model = AutoModelForSequenceClassification.from_pretrained(
114
  "hun3359/klue-bert-base-sentiment", cache_dir=CACHE_DIR
115
  )
116
  model.eval()
117
- return tok, model
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():