Update README.md
Browse files
README.md
CHANGED
@@ -60,7 +60,7 @@ from transformers import HubertForSequenceClassification, Wav2Vec2FeatureExtract
|
|
60 |
|
61 |
|
62 |
# CONFIG and MODEL SETUP
|
63 |
-
model_name = '
|
64 |
feature_extractor = Wav2Vec2FeatureExtractor.from_pretrained("facebook/hubert-base-ls960")
|
65 |
model = HubertForSequenceClassification.from_pretrained(model_name)
|
66 |
model.classifier = nn.Linear(in_features=256,out_features=6)
|
|
|
60 |
|
61 |
|
62 |
# CONFIG and MODEL SETUP
|
63 |
+
model_name = 'amiriparian/HuBERT-EmoSet'
|
64 |
feature_extractor = Wav2Vec2FeatureExtractor.from_pretrained("facebook/hubert-base-ls960")
|
65 |
model = HubertForSequenceClassification.from_pretrained(model_name)
|
66 |
model.classifier = nn.Linear(in_features=256,out_features=6)
|