Update app.py

app.py

@@ -6,7 +6,7 @@ gpt2 = pipeline('text-generation', model = "Lifan-Z/Chinese-Classic-Poem-Generat
 
 def poems(st):
   poems=""
-  sequences = gpt2('<|endoftext|>'+st, max_length=50, do_sample=True, top_k=950, repetition_penalty=1.2, num_return_sequences=20, eos_token_id=0)
+  sequences = gpt2('<|endoftext|>'+st, max_length=50, do_sample=True, top_k=20, top_p=0.9, repetition_penalty=1.2, num_return_sequences=20, eos_token_id=0)
   i=0
   for seq in sequences:
     if (len(seq.get('generated_text')) == 108):