This wav2vec2 based emotion detection model is trained on the emo-DB dataset. Code for training can be found here.
Emotion classes are:
- 0: 'anger'
- 1: 'boredom'
- 2: 'disgust'
- 3: 'fear'
- 4: 'happiness'
- 5: 'sadness'
- 6: 'neutral'
inference.py shows, how the model can be used.
- Downloads last month
- 379