xbgoose commited on
Commit
040aa13
1 Parent(s): 01fec04

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -45,7 +45,7 @@ import torchaudio
45
  import torch
46
 
47
  feature_extractor = Wav2Vec2FeatureExtractor.from_pretrained("facebook/hubert-large-ls960-ft")
48
- model = HubertForSequenceClassification.from_pretrained("xbgoose/hubert-dusha-finetuned")
49
  num2emotion = {0: 'neutral', 1: 'angry', 2: 'positive', 3: 'sad', 4: 'other'}
50
 
51
  filepath = "path/to/audio.wav"
 
45
  import torch
46
 
47
  feature_extractor = Wav2Vec2FeatureExtractor.from_pretrained("facebook/hubert-large-ls960-ft")
48
+ model = HubertForSequenceClassification.from_pretrained("xbgoose/hubert-speech-emotion-recognition-russian-dusha-finetuned")
49
  num2emotion = {0: 'neutral', 1: 'angry', 2: 'positive', 3: 'sad', 4: 'other'}
50
 
51
  filepath = "path/to/audio.wav"