Update README.md

README.md

@@ -24,11 +24,23 @@ pipeline_tag: text-classification
 
 This is a RoBERTa-based model fine-tuned on the Empathetic Dialogues dataset for conversational emotion classification. The model leverages the powerful RoBERTa architecture to understand and classify emotional contexts in conversational text.
 
+### Emotion Classes
+The model is trained to classify conversations into the following emotional categories:
+- Surprised
+- Angry
+- Sad
+- Joyful
+- Anxious
+- Hopeful
+- Confident
+- Disappointed
+
 ### Model Details
 - **Base Model**: roberta-base
 - **Task**: Emotion Classification in Conversations
 - **Dataset**: Empathetic Dialogues
 - **Training Approach**: Full Fine-Tuning
+- **Number of Emotion Classes**: 8
 
 ### Model Performance
 
@@ -84,7 +96,8 @@ with torch.no_grad():
 
 ## Limitations
 - Performance may vary with out-of-domain conversational contexts
-- Relies on the specific emotional categories in the Empathetic Dialogues dataset
+- Emotion classification limited to the 8 specified emotional categories
+- Relies on the specific emotional nuances in the Empathetic Dialogues dataset
 - Requires careful interpretation in real-world applications
 
 ## Ethical Considerations