|
|
--- |
|
|
datasets: |
|
|
- AiLab-IMCS-UL/twitter_emotions-en |
|
|
base_model: |
|
|
- distilbert/distilbert-base-uncased |
|
|
pipeline_tag: text-classification |
|
|
--- |
|
|
|
|
|
# π Summary |
|
|
emotion classification model with distilbert |
|
|
|
|
|
# π Quick start (Python) |
|
|
|
|
|
### Installation |
|
|
```bash |
|
|
pip install transformers torch |
|
|
``` |
|
|
|
|
|
### Basic Usage |
|
|
```python |
|
|
from transformers import pipeline |
|
|
|
|
|
classifier = pipeline("text-classification", model="yoon-eunbin/distilbert-emotion-classification-model") |
|
|
result = classifier("I'm so happy today!") |
|
|
print(result) |
|
|
``` |
|
|
|