hhou435 commited on
Commit
a51fdec
1 Parent(s): 99fb15c

Update README

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -26,7 +26,7 @@ When the parameter skip_special_tokens is True:
26
  >>> model = GPT2LMHeadModel.from_pretrained("uer/gpt2-chinese-poem")
27
  >>> text_generator = TextGenerationPipeline(model, tokenizer)
28
  >>> text_generator("[CLS]梅 山 如 积 翠 ,", max_length=50, do_sample=True)
29
- [{'generated_text': '[CLS]梅 山 如 积 翠 , '}]
30
  ```
31
 
32
  When the parameter skip_special_tokens is False:
@@ -36,8 +36,8 @@ When the parameter skip_special_tokens is False:
36
  >>> tokenizer = BertTokenizer.from_pretrained("uer/gpt2-chinese-poem")
37
  >>> model = GPT2LMHeadModel.from_pretrained("uer/gpt2-chinese-poem")
38
  >>> text_generator = TextGenerationPipeline(model, tokenizer)
39
- >>> text_generator("[CLS]梅 山 如 积 翠 ,", max_length=50, do_sample=True)
40
- [{'generated_text': '[CLS]梅 山 如 积 翠 , [UNK] , 石 来 , '}]
41
  ```
42
 
43
  ## Training data
 
26
  >>> model = GPT2LMHeadModel.from_pretrained("uer/gpt2-chinese-poem")
27
  >>> text_generator = TextGenerationPipeline(model, tokenizer)
28
  >>> text_generator("[CLS]梅 山 如 积 翠 ,", max_length=50, do_sample=True)
29
+ [{'generated_text': '[CLS]梅 山 如 积 翠 , 向 湖 边 过 , 偏 怜 雪 里 看 。 浮 峦 如 画 出 , 远 树 与 天 连 。 月 上 僧 房 静 , 风 回 萤 火 寒 。 幽 情 何 可 写 , 赖 有 子 期 弹 。 棠 真'}]
30
  ```
31
 
32
  When the parameter skip_special_tokens is False:
 
36
  >>> tokenizer = BertTokenizer.from_pretrained("uer/gpt2-chinese-poem")
37
  >>> model = GPT2LMHeadModel.from_pretrained("uer/gpt2-chinese-poem")
38
  >>> text_generator = TextGenerationPipeline(model, tokenizer)
39
+ >>> text_generator("[CLS]梅 山 如 积 翠 ,", max_length=100, do_sample=True)
40
+ [{'generated_text': '[CLS]梅 山 如 积 翠 , 枝 高 青 云 蒙 。 常 因 风 露 晚 , 隠 映 瑶 台 中 。 忽 闻 山 吹 天 风 高 , 迥 出 凡 境 空 。 清 影 落 潭 水 , 暗 香 逈 峰 。 却 寻 白 太 白 [SEP] 而 非'}]
41
  ```
42
 
43
  ## Training data