Lifan-Z's picture
Update README.md
6fe8a44
|
raw
history blame
2.11 kB
metadata
license: apache-2.0
language:
  - zh
tags:
  - art

How to use "Lifan-Z/Chinese-Classic-Poem-Generator-style5x8-GPT2":

# The link to the demo is at the bottom right corner of the Model Card.

import transformers
from transformers import pipeline

gpt2 = pipeline('text-generation', model = "Lifan-Z/Chinese-Classic-Poem-Generator-style5x8-GPT2")
sequences = gpt2('<|endoftext|>雨', max_length=50, do_sample=True, top_k=20, top_p=0.9, repetition_penalty=1.2, num_return_sequences=6, eos_token_id=0)
for seq in sequences:
    print(seq)

Output:
{'generated_text': '<|endoftext|>雨 过 山 声 远 , 潮 来 水 势 平 。 风 和 渔 父 起 , 日 澹 草 堂 明 。 月 上 云 添 画 , 帆 移 浪 卷 城 。 无 言 归 路 好 , 更 觉 旧 时 情 。'}
{'generated_text': '<|endoftext|>雨 后 山 中 景 , 春 来 客 里 心 。 风 清 莺 不 语 , 云 散 鸟 还 眠 。 落 照 摇 红 酒 , 残 阳 照 紫 泥 。 故 人 无 此 趣 , 相 送 又 东 寻 。'}
{'generated_text': '<|endoftext|>雨 余 天 影 半 , 路 僻 出 阶 除 。 树 矗 山 扉 重 , 溪 泠 水 汽 清 。 耕 夫 齐 舍 逐 , 兵 卒 峻 庭 隆 。 何 必 投 吾 圃 , 寻 根 傍 石 林 。'}
{'generated_text': '<|endoftext|>雨 足 风 声 恶 , 秋 深 雾 气 清 。 野 花 生 未 谢 , 江 雁 度 还 行 。 落 日 山 前 水 , 残 阳 树 下 云 。 明 朝 去 留 远 , 此 去 一 沾 巾 。'}
{'generated_text': '<|endoftext|>雨 歇 山 容 合 , 江 晴 海 气 和 。 一 家 依 水 市 , 三 径 入 云 林 。 风 物 今 如 许 , 天 情 老 似 绵 。 清 溪 无 限 意 , 聊 与 问 归 寻 。'}
{'generated_text': '<|endoftext|>雨 过 水 多 处 , 林 开 竹 自 阴 。 石 穿 山 路 滑 , 树 乱 涧 声 深 。 鸟 啄 泥 沙 溜 , 人 归 草 木 丛 。 平 生 一 樽 酒 , 何 日 重 相 寻 。'}

References:

The model uploaded to HuggingFace: "yuanzhoulvpi/gpt2_chinese"
https://github.com/chinese-poetry/chinese-poetry/tree/master/%E5%85%A8%E5%94%90%E8%AF%97