m3hrdadfi
/

wav2vec2-base-100k-gtzan-music-genres

Automatic Speech Recognition

audio-classification

Inference Endpoints

Model card Files Files and versions Community

m3hrdadfi commited on Jun 6, 2021

Commit

15a5eff

·

1 Parent(s): 9224021

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -39,7 +39,7 @@ import pandas as pd
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model_name_or_path = "m3hrdadfi/wav2vec2-base-100k-voxpopuli-gtzan-music"
 config = AutoConfig.from_pretrained(model_name_or_path)
-feature_extractor = Wav2Vec2Processor.from_pretrained(model_name_or_path)
 sampling_rate = feature_extractor.sampling_rate
 model = Wav2Vec2ForSpeechClassification.from_pretrained(model_name_or_path).to(device)
 ```

 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model_name_or_path = "m3hrdadfi/wav2vec2-base-100k-voxpopuli-gtzan-music"
 config = AutoConfig.from_pretrained(model_name_or_path)
+feature_extractor = Wav2Vec2FeatureExtractor.from_pretrained(model_name_or_path)
 sampling_rate = feature_extractor.sampling_rate
 model = Wav2Vec2ForSpeechClassification.from_pretrained(model_name_or_path).to(device)
 ```