duault
commited on
Update README.md
Browse files
README.md
CHANGED
@@ -27,8 +27,8 @@ from transformers import Wav2Vec2ForSequenceClassification, Wav2Vec2FeatureExtra
|
|
27 |
import torch
|
28 |
|
29 |
# Load model and feature extractor
|
30 |
-
model = Wav2Vec2ForSequenceClassification.from_pretrained("
|
31 |
-
feature_extractor = Wav2Vec2FeatureExtractor.from_pretrained("
|
32 |
|
33 |
# Process audio file
|
34 |
audio_path = "path/to/audio.wav"
|
|
|
27 |
import torch
|
28 |
|
29 |
# Load model and feature extractor
|
30 |
+
model = Wav2Vec2ForSequenceClassification.from_pretrained("gastoooon/music-classifier")
|
31 |
+
feature_extractor = Wav2Vec2FeatureExtractor.from_pretrained("facebook/wav2vec2-large")
|
32 |
|
33 |
# Process audio file
|
34 |
audio_path = "path/to/audio.wav"
|