amiriparian commited on
Commit
219d8ee
1 Parent(s): 54ec632

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -60,7 +60,7 @@ from transformers import HubertForSequenceClassification, Wav2Vec2FeatureExtract
60
 
61
 
62
  # CONFIG and MODEL SETUP
63
- model_name = '.../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)
 
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)