Update README.md
Browse files
README.md
CHANGED
|
@@ -120,12 +120,12 @@ print(result)
|
|
| 120 |
```python
|
| 121 |
from transformers import pipeline
|
| 122 |
|
| 123 |
-
|
| 124 |
|
| 125 |
classifier = pipeline(
|
| 126 |
"text-classification",
|
| 127 |
-
model=
|
| 128 |
-
tokenizer=
|
| 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 |
|