Update README.md
Browse files
README.md
CHANGED
@@ -13,7 +13,7 @@ import transformers
|
|
13 |
from transformers import pipeline
|
14 |
|
15 |
gpt2 = pipeline('text-generation', model = "Lifan-Z/Chinese-Classic-Poem-Generator-style5x8-GPT2")
|
16 |
-
sequences = gpt2('<|endoftext|>雨', max_length=50, num_return_sequences=6, eos_token_id=0)
|
17 |
for seq in sequences:
|
18 |
print(seq)
|
19 |
|
|
|
13 |
from transformers import pipeline
|
14 |
|
15 |
gpt2 = pipeline('text-generation', model = "Lifan-Z/Chinese-Classic-Poem-Generator-style5x8-GPT2")
|
16 |
+
sequences = gpt2('<|endoftext|>雨', max_length=50, do_sample=True, top_k=20, top_p=0.9, num_return_sequences=6, eos_token_id=0)
|
17 |
for seq in sequences:
|
18 |
print(seq)
|
19 |
|