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
- 190
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social
visibility and check back later, or deploy to Inference Endpoints (dedicated)
instead.