saurabhati commited on
Commit
fd4f018
·
verified ·
1 Parent(s): c019c1e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -44,9 +44,9 @@ import torch
44
  import librosa
45
  from transformers import AutoConfig, AutoModelForAudioClassification, AutoFeatureExtractor
46
 
47
- config = AutoConfig.from_pretrained('saurabhati/DASS_small_47.2',trust_remote_code=True)
48
- audio_model = AutoModelForAudioClassification.from_pretrained('saurabhati/DASS_small_47.2',trust_remote_code=True)
49
- feature_extractor = AutoFeatureExtractor.from_pretrained('saurabhati/DASS_small_47.2',trust_remote_code=True)
50
 
51
  waveform, sr = librosa.load("audio/eval/_/_/--4gqARaEJE_0.000.flac", sr=16000)
52
  inputs = feature_extractor(waveform,sr, return_tensors='pt')
 
44
  import librosa
45
  from transformers import AutoConfig, AutoModelForAudioClassification, AutoFeatureExtractor
46
 
47
+ config = AutoConfig.from_pretrained('saurabhati/DASS_small_AudioSet_47.2',trust_remote_code=True)
48
+ audio_model = AutoModelForAudioClassification.from_pretrained('saurabhati/DASS_small_AudioSet_47.2',trust_remote_code=True)
49
+ feature_extractor = AutoFeatureExtractor.from_pretrained('saurabhati/DASS_small_AudioSet_47.2',trust_remote_code=True)
50
 
51
  waveform, sr = librosa.load("audio/eval/_/_/--4gqARaEJE_0.000.flac", sr=16000)
52
  inputs = feature_extractor(waveform,sr, return_tensors='pt')