Lifan-Z's picture
Update README.md
9606694 verified
metadata
license: apache-2.0
language:
  - zh
tags:
  - art

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