Update README.md
Browse files
README.md
CHANGED
@@ -20,7 +20,7 @@ widget:
|
|
20 |
|
21 |
## Description
|
22 |
|
23 |
-
The model was trained using the BERT language model on
|
24 |
For training and test codes check out [Github page](https://github.com/Veucci/lyrics-to-genre-lite).
|
25 |
|
26 |
## Dataset
|
|
|
20 |
|
21 |
## Description
|
22 |
|
23 |
+
The model was trained using the BERT language model on my [song lyrics dataset](https://huggingface.co/datasets/Veucci/lyrics_3genre) to predict the genre of a given song based on its lyrics. This repository houses the machine learning model, which is capable of making predictions in three distinct genres: Pop, Rock, and Hip-Hop.
|
24 |
For training and test codes check out [Github page](https://github.com/Veucci/lyrics-to-genre-lite).
|
25 |
|
26 |
## Dataset
|