Downloads · 30 days
11
8% of all-time downloads
KEERTHANA6805/bert-emotion
bert-emotion is a text classification model from KEERTHANA6805. Use it when you need a label for a piece of text. It is set up for transformers. The card lists the license as mit.
This model classifies text into 6 emotion categories: - joy - sadness - anger - fear - love - surprise
Downloads · 30 days
11
8% of all-time downloads
All-time downloads
139
Public
Parameters
109M
438 MB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors438 MB · 100%
From the Hugging Face model README
This model classifies text into 6 emotion categories:
Model: bert-base-uncased
Dataset: emotion (Hugging Face datasets library)
from transformers import pipeline
classifier = pipeline("text-classification", model="KEERTHANA6805/bert-emotion")
classifier("I am so happy today!")