Lifan-Z

/

Chinese-Classic-Poem-Generator-style5x4-GPT2

like 0

How to use "Lifan-Z/Chinese-Classic-Poem-Generator-style5x4-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-style5x4-GPT2")  
sequences = gpt2('<|endoftext|>雨', max_length=26, do_sample=True, top_k=20, top_p=0.9, 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