speechbrain
/

emotion-recognition-wav2vec2-IEMOCAP

Audio Classification

Model card Files Files and versions Community

Mirco commited on Oct 20, 2021

Commit

59c6806

•

1 Parent(s): 656ec10

Update README.md

Files changed (1) hide show

README.md +2 -3

README.md CHANGED Viewed

@@ -53,11 +53,10 @@ Please notice that we encourage you to read our tutorials and learn more about
 ### Perform Emotion recognition
 ```python
-import torchaudio
 from speechbrain.pretrained.interfaces import EncoderWav2vecClassifier
 classifier = EncoderWav2vecClassifier.from_hparams(source="speechbrain/emotion-recognition-wav2vec2-IEMOCAP")
-signal, fs =torchaudio.load('/workspace/emotion-recognition-wav2vec2/anger.wav')
-prediction = classifier.classify_batch(sig)
 ```
  The prediction tensor will contain a tuple of (embedding, id_class, label_name).

 ### Perform Emotion recognition
 ```python
 from speechbrain.pretrained.interfaces import EncoderWav2vecClassifier
 classifier = EncoderWav2vecClassifier.from_hparams(source="speechbrain/emotion-recognition-wav2vec2-IEMOCAP")
+out_prob, score, index, text_lab = classifier.classify_file("speechbrain/emotion-recognition-wav2vec2-IEMOCAP/anger.wav")
+print(text_lab)
 ```
  The prediction tensor will contain a tuple of (embedding, id_class, label_name).