LimYeri commited on
Commit
031e069
·
verified ·
1 Parent(s): 1ffbc3e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -120,12 +120,12 @@ print(result)
120
  ```python
121
  from transformers import pipeline
122
 
123
- model_name = "LimYeri/HowRU-KoELECTRA-Emotion-Classifier"
124
 
125
  classifier = pipeline(
126
  "text-classification",
127
- model=model_name,
128
- tokenizer=model_name,
129
  top_k=None # 전체 감정 확률 반환
130
  )
131
 
 
120
  ```python
121
  from transformers import pipeline
122
 
123
+ MODEL_NAME = "LimYeri/HowRU-KoELECTRA-Emotion-Classifier"
124
 
125
  classifier = pipeline(
126
  "text-classification",
127
+ model=MODEL_NAME,
128
+ tokenizer=MODEL_NAME,
129
  top_k=None # 전체 감정 확률 반환
130
  )
131