Update README.md
Browse files
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/
|
| 25 |
-
audio_model = AutoModelForAudioClassification.from_pretrained('saurabhati/
|
| 26 |
-
feature_extractor = AutoFeatureExtractor.from_pretrained('saurabhati/
|
| 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')
|