Update README.md

README.md

@@ -15,22 +15,31 @@ should probably proofread and complete it, then remove this comment. -->
 
 # emotions_6_classes_small
 
-This model is a fine-tuned version of [facebook/wav2vec2-base](https://huggingface.co/facebook/wav2vec2-base) on the None dataset.
+This model is a fine-tuned version of [facebook/wav2vec2-base](https://huggingface.co/facebook/wav2vec2-base) on the 'Audio emotions' public dataset, available form https://www.kaggle.com/datasets/uldisvalainis/audio-emotions.
+'Surprised' class was discarded due to lack of samples.
+
 It achieves the following results on the evaluation set:
 - Loss: 0.9106
 - Accuracy: 0.7920
 
 ## Model description
 
-More information needed
+Classifies audios into 6 emotions:
+- Angry
+- Happy
+- Sad
+- Neutral
+- Fearful
+- Disgusted
 
 ## Intended uses & limitations
 
-More information needed
+This model was trained for educational purposes.
 
 ## Training and evaluation data
 
-More information needed
+- Training: 80%
+- Test: 20%
 
 ## Training procedure