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: 34

Inference Examples

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.