audeering
/

wav2vec2-large-robust-12-ft-emotion-msp-dim

Audio Classification

emotion-recognition

Inference Endpoints

Model card Files Files and versions Community

audmax commited on Jul 8

Commit

1ea7d89

•

1 Parent(s): fefff98

Fix for device='cuda'

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -85,7 +85,7 @@ class EmotionModel(Wav2Vec2PreTrainedModel):
 device = 'cpu'
 model_name = 'audeering/wav2vec2-large-robust-12-ft-emotion-msp-dim'
 processor = Wav2Vec2Processor.from_pretrained(model_name)
-model = EmotionModel.from_pretrained(model_name)
 # dummy signal
 sampling_rate = 16000

 device = 'cpu'
 model_name = 'audeering/wav2vec2-large-robust-12-ft-emotion-msp-dim'
 processor = Wav2Vec2Processor.from_pretrained(model_name)
+model = EmotionModel.from_pretrained(model_name).to(device)
 # dummy signal
 sampling_rate = 16000