JeffreyLau commited on
Commit
c71020c
1 Parent(s): 4297407

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -35,7 +35,7 @@ When the parameter skip_special_tokens is False:
35
 
36
  ```python
37
  >>> from transformers import BertTokenizer, GPT2LMHeadModel,TextGenerationPipeline
38
- >>> tokenizer = BertTokenizer.from_pretrained("uer/gpt2-chinese-poem")
39
  >>> model = GPT2LMHeadModel.from_pretrained("JeffreyLau/SikuGPT2-poem")
40
  >>> text_generator = TextGenerationPipeline(model, tokenizer)
41
  >>> text_generator("[CLS]梅 山 如 积 翠 ,", max_length=100, do_sample=True)
 
35
 
36
  ```python
37
  >>> from transformers import BertTokenizer, GPT2LMHeadModel,TextGenerationPipeline
38
+ >>> tokenizer = BertTokenizer.from_pretrained("JeffreyLau/SikuGPT2-poem")
39
  >>> model = GPT2LMHeadModel.from_pretrained("JeffreyLau/SikuGPT2-poem")
40
  >>> text_generator = TextGenerationPipeline(model, tokenizer)
41
  >>> text_generator("[CLS]梅 山 如 积 翠 ,", max_length=100, do_sample=True)