Update README.md

README.md

@@ -25,9 +25,9 @@ The lyric creation robot is called “小L”. So at the end of the input questi
 You can use this model directly with a pipeline for text2text generation:
 
 ```python
->>> from transformers import BertTokenizer, T5ForConditionalGeneration, Text2TextGenerationPipeline
->>> tokenizer = BertTokenizer.from_pretrained("uer/t5-small-chinese-cluecorpussmall")
+>>> from transformers import T5Tokenizer, T5ForConditionalGeneration, Text2TextGenerationPipeline
+>>> tokenizer = T5Tokenizer.from_pretrained("souljoy/t5-chinese-lyric")
 >>> model = T5ForConditionalGeneration.from_pretrained("souljoy/t5-chinese-lyric")
 >>> text2text_generator = Text2TextGenerationPipeline(model, tokenizer)  
->>> text2text_generator("用户：写一首歌，歌手“毛毛”，歌曲“多少年后”，以“谎言，伤痛，心爱，掩饰，退一步，寻找”为主题。\\n小L：", max_length=512, do_sample=False)
+>>> text2text_generator("用户：写一首歌，歌手“毛毛”，歌曲“多少年后”，以“谎言，伤痛，心爱，掩饰，退一步，寻找”为主题。\\n小L：", max_length=512, do_sample=True)
 ```