Update README.md

README.md

@@ -7,11 +7,11 @@ widget:
 ---
 
 
-# Chinese GPT2 Model
+# Chinese GPT2 Lyric Model
 
 ## Model description
 
-The model is used to generate Chinese lyrics. You can download the model  either from the [GPT2-Chinese Github page](https://github.com/Morizeyao/GPT2-Chinese), or via HuggingFace from the link [gpt2-chinese-lyric](https://huggingface.co/uer/gpt2-chinese-lyric)
+The model is used to generate Chinese lyrics. You can download the model either from the [GPT2-Chinese Github page](https://github.com/Morizeyao/GPT2-Chinese), or via HuggingFace from the link [gpt2-chinese-lyric](https://huggingface.co/uer/gpt2-chinese-lyric)
 
 ## How to use
 
@@ -26,11 +26,9 @@ You can use the model directly with a pipeline for text generation:
     [{'generated_text': '最美的不是下雨天，是曾与你躲过雨的屋檐 ， 下 课 铃 声 响 起 的 瞬 间 ， 我 们 的 笑 脸 ， 有 太 多 回 忆 在 浮 现 ， 是 你 总 在 我 身 边 ， 不 知 道 会 不 会 再 见 ， 从 现 在 开 始 到 永 远 ， 想 说 的 语 言 凝 结 成 一 句 ， 不 管 我 们 是 否 能 够 兑 现 ， 想 说 的 语 言 凝 结'}]
 ```
 
-
-
 ## Training data
 
-Training data contains 150,000 Chinese lyrics which are collected by [Chinese-Lyric-Corpus](https://github.com/gaussic/Chinese-Lyric-Corpus) and [MusicLyricChatbot](https://github.com/liuhuanyong/MusicLyricChatbot) projects
+Training data contains 150,000 Chinese lyrics which are collected by [Chinese-Lyric-Corpus](https://github.com/gaussic/Chinese-Lyric-Corpus) and [MusicLyricChatbot](https://github.com/liuhuanyong/MusicLyricChatbot).
 
 ## Training procedure
 
@@ -39,12 +37,12 @@ The model is pre-trained by [UER-py](https://github.com/dbiir/UER-py/) on [Tence
 ```
 python3 preprocess.py --corpus_path corpora/lyric.txt \
                       --vocab_path models/google_zh_vocab.txt \
-                      --dataset_path lyric_lm_seq512_dataset.pt \
-                      --seq_length 512 --processes_num 32 --target lm 
+                      --dataset_path lyric_dataset.pt --processes_num 32 \
+                      --seq_length 512 --target lm 
 ```
 
 ```
-python3 pretrain.py --dataset_path lyric_lm_seq512_dataset.pt \
+python3 pretrain.py --dataset_path lyric_dataset.pt \
 					--pretrained_model_path gpt2-base-chinese-cluecorpussmall/pytorch_model.bin\
                     --vocab_path models/google_zh_vocab.txt \
                     --output_model_path models/lyric_gpt2_seq512_model.bin \