gastonduault
/

music-classifier

Audio Classification

Model card Files Files and versions Community

gastonduault commited on Nov 17, 2024

Commit

4365ba4

·

verified ·

1 Parent(s): 445fbef

Update predict-example.py

Files changed (1) hide show

predict-example.py +1 -1

predict-example.py CHANGED Viewed

@@ -21,7 +21,7 @@ for example in dataset["train"]:
             break
 print(f"Loading model from {MODEL_DIR}...\n")
-model = Wav2Vec2ForSequenceClassification.from_pretrained("gastoooon/music-classifier")
 feature_extractor = Wav2Vec2FeatureExtractor.from_pretrained("facebook/wav2vec2-large")
 # Function for preprocessing audio for prediction

             break
 print(f"Loading model from {MODEL_DIR}...\n")
+model = Wav2Vec2ForSequenceClassification.from_pretrained("gastonduault/music-classifier")
 feature_extractor = Wav2Vec2FeatureExtractor.from_pretrained("facebook/wav2vec2-large")
 # Function for preprocessing audio for prediction