skytnt
/

gpt2-japanese-lyric-small

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

skytnt commited on May 8, 2022

Commit

b43899d

•

1 Parent(s): 016b4b4

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -22,7 +22,7 @@ The model is used to generate Japanese lyrics.
 ```python
 import torch
-from transformers import T5Tokenizer, GPT2LMHeadModel, TextGenerationPipeline
 tokenizer = T5Tokenizer.from_pretrained("skytnt/gpt2-japanese-lyric-small")
 model = GPT2LMHeadModel.from_pretrained("skytnt/gpt2-japanese-lyric-small")
@@ -64,5 +64,5 @@ print(gen_lyric("未来に揺れる花 過去にもあった花"))
 ## Training data
-Training data ([click to download](https://data.anyweb.xyz/dataset/lyric.zip)) contains 48,394 Japanese lyrics which are collected from [NetEasyMusic](https://music.163.com/) by [lyric_download](https://github.com/SkyTNT/lyric_downlowd)

 ```python
 import torch
+from transformers import T5Tokenizer, GPT2LMHeadModel
 tokenizer = T5Tokenizer.from_pretrained("skytnt/gpt2-japanese-lyric-small")
 model = GPT2LMHeadModel.from_pretrained("skytnt/gpt2-japanese-lyric-small")
 ## Training data
+[Training data](https://data.anyweb.xyz/dataset/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)