gastonduault
/

music-classifier

Audio Classification

Model card Files Files and versions

gastonduault commited on Nov 22, 2024

Commit

f55b848

·

verified ·

1 Parent(s): 830dc58

Update README.md

Files changed (1) hide show

README.md +2 -1

README.md CHANGED Viewed

@@ -15,7 +15,8 @@ pipeline_tag: audio-classification
 ---
 # Music Genre Classification Model 🎶
-This model classifies music genres based on audio signals. It was fine-tuned on the `music_genres_small` dataset using the Wav2Vec2 architecture.
 You can find a GitHub repository with an interface hosted by a Flask API to test the model:  **[music-classifier repository](https://github.com/gastonduault/Music-Classifier)**
 ## Metrics

 ---
 # Music Genre Classification Model 🎶
+This model classifies music genres based on audio signals.
+It was fine-tuned on the model  **[Wav2Vec2](https://huggingface.co/facebook/wav2vec2-large)** and using the datasets **[music_genres_small](https://huggingface.co/datasets/lewtun/music_genres_small)**.
 You can find a GitHub repository with an interface hosted by a Flask API to test the model:  **[music-classifier repository](https://github.com/gastonduault/Music-Classifier)**
 ## Metrics