Update README.md

README.md

@@ -20,11 +20,12 @@ widget:
 
 ## Description
 
-This repository contains a machine learning model trained on a song lyrics dataset to predict the genre of a given song based on its lyrics. The model is capable of making predictions in three different genres: Pop, Rock, and Hip-Hop.
+The model was trained using the BERT language model on a song lyrics dataset 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.
+For training and test codes check out [Github page](https://github.com/Veucci/lyrics-to-genre-lite).
 
 ## Dataset
 
-The model was trained on a diverse and labeled dataset of song lyrics. The dataset consists of songs from various artists and genres, carefully curated to provide a comprehensive representation of Pop, Rock, and Hip-Hop music.
+The model was trained on a diverse and labeled dataset of song lyrics, which contained approximately 3000 rows. The dataset was carefully curated to include songs from a wide range of artists and genres, ensuring a comprehensive representation of Pop, Rock, and Hip-Hop music.
 [DATASET](https://huggingface.co/datasets/Veucci/lyrics_3genre)
 
 ## License