YAML Metadata Warning:empty or missing yaml metadata in repo card

Check out the documentation for more information.

🧠 DistilBERT Emotion Classification (by Subhra Sikha Sahoo)

This model is a fine-tuned version of distilbert-base-uncased for emotion classification.
It predicts the emotion expressed in a given text such as joy, sadness, anger, fear, love, surprise, etc.


πŸ“˜ Model Details

  • Base Model: distilbert-base-uncased
  • Trained By: Subhra Sikha Sahoo
  • Task: Multi-label Emotion Classification
  • Language: English
  • Dataset Size: 20,000 samples

πŸš€ Usage Example

from transformers import pipeline

pipe = pipeline("text-classification", model="Subhra-123/distilbert-base-emotion", return_all_scores=True)
text = "I love my friends!"
pred = pipe(text)[0]
emotion = max(pred, key=lambda x: x['score'])
print(f"{text} β†’ {emotion['label']} ({emotion['score']:.2f})")
Downloads last month
5
Safetensors
Model size
67M params
Tensor type
F32
Β·
Inference Providers NEW
This model isn't deployed by any Inference Provider. πŸ™‹ Ask for provider support

Space using Subhra-123/distilbert-base-emotion 1