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.