amiriparian commited on
Commit
f4f5ec5
1 Parent(s): b158d45

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -62,7 +62,7 @@ from transformers import AutoModelForAudioClassification, Wav2Vec2FeatureExtract
62
  # CONFIG and MODEL SETUP
63
  model_name = 'amiriparian/HuBERT-EmoSet'
64
  feature_extractor = Wav2Vec2FeatureExtractor.from_pretrained("facebook/hubert-base-ls960")
65
- AutoModelForAudioClassification.from_pretrained(model_name, trust_remote_code=True)
66
  model.freeze_og_encoder()
67
 
68
  sampling_rate=16000
 
62
  # CONFIG and MODEL SETUP
63
  model_name = 'amiriparian/HuBERT-EmoSet'
64
  feature_extractor = Wav2Vec2FeatureExtractor.from_pretrained("facebook/hubert-base-ls960")
65
+ model = AutoModelForAudioClassification.from_pretrained(model_name, trust_remote_code=True,revision="b158d45ed8578432468f3ab8d46cbe5974380812")
66
  model.freeze_og_encoder()
67
 
68
  sampling_rate=16000