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