Update README.md
Browse files
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"
|