saurabhati commited on
Commit
89a2d30
·
verified ·
1 Parent(s): 8a38d16

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -21,9 +21,9 @@ import torch
21
  import librosa
22
  from transformers import AutoConfig, AutoModelForAudioClassification, AutoFeatureExtractor
23
 
24
- config = AutoConfig.from_pretrained('saurabhati/DASS_small_AudioSet_50.2',trust_remote_code=True)
25
- audio_model = AutoModelForAudioClassification.from_pretrained('saurabhati/DASS_small_AudioSet_50.2',trust_remote_code=True)
26
- feature_extractor = AutoFeatureExtractor.from_pretrained('saurabhati/DASS_small_AudioSet_50.2',trust_remote_code=True)
27
 
28
  waveform, sr = librosa.load("audio/eval/_/_/--4gqARaEJE_0.000.flac", sr=16000)
29
  inputs = feature_extractor(waveform,sr, return_tensors='pt')
 
21
  import librosa
22
  from transformers import AutoConfig, AutoModelForAudioClassification, AutoFeatureExtractor
23
 
24
+ config = AutoConfig.from_pretrained('saurabhati/DASS_medium_AudioSet_50.2',trust_remote_code=True)
25
+ audio_model = AutoModelForAudioClassification.from_pretrained('saurabhati/DASS_medium_AudioSet_50.2',trust_remote_code=True)
26
+ feature_extractor = AutoFeatureExtractor.from_pretrained('saurabhati/DASS_medium_AudioSet_50.2',trust_remote_code=True)
27
 
28
  waveform, sr = librosa.load("audio/eval/_/_/--4gqARaEJE_0.000.flac", sr=16000)
29
  inputs = feature_extractor(waveform,sr, return_tensors='pt')