Update README.md

README.md

@@ -9,7 +9,9 @@ tags:
 - nlp
 license: mit
 widget:
-- text: "<s>桜[CLS]"
+- text: <s>桜[CLS]
+datasets:
+- skytnt/japanese-lyric
 ---
 
 # Japanese GPT2 Lyric Model
@@ -76,5 +78,4 @@ print(gen_lyric("桜",""))
 
 ## Training data
 
-[Training data](https://github.com/SkyTNT/gpt2-japanese-lyric/blob/main/lyric_ids_titled.pkl) contains 143,587 Japanese lyrics which are collected from [uta-net](https://www.uta-net.com/) by [lyric_download](https://github.com/SkyTNT/lyric_downlowd)
-
+[Training data](https://huggingface.co/datasets/skytnt/japanese-lyric/blob/main/lyric_clean.pkl) contains 143,587 Japanese lyrics which are collected from [uta-net](https://www.uta-net.com/) by [lyric_download](https://github.com/SkyTNT/lyric_downlowd)