Upload README.md with huggingface_hub

README.md

@@ -33,7 +33,7 @@ This model uses the DistilBERT architecture, which is a lighter and faster varia
 ### Training Dataset
 - **Dataset:** [dair-ai/emotion](https://huggingface.co/datasets/dair-ai/emotion)
 - **Training Set Size:** 16,000 examples
-- **Dataset Description:** The dataset contains English sentences labeled with six emotional categories: anger, joy, optimism, sadness, fear, and disgust.
+- **Dataset Description:** The dataset contains English sentences labeled with six emotional categories: sadness (0), joy (1), love (2), anger (3), fear (4), surprise (5)
 
 ### Results
 - **Training Time:** ~226 seconds