Update README
Browse files
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=
|
40 |
-
[{'generated_text': '[CLS]梅 山 如 积 翠 ,
|
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
|