Lifan-Z
/

Chinese-Classic-Poem-Generator-style5x4-GPT2

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

Lifan-Z commited on Jan 29

Commit

9606694

•

1 Parent(s): 0ceb639

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -13,7 +13,7 @@ tags:
 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, num_return_sequences=6, eos_token_id=0)
 for seq in sequences:
     print(seq)

 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)