diff --git a/generator/gpt2/gpt2_generator.py b/generator/gpt2/gpt2_generator.py
index ab2bb11..3b9ce22 100644
--- a/generator/gpt2/gpt2_generator.py
+++ b/generator/gpt2/gpt2_generator.py
@@ -11,7 +11,7 @@ tf.compat.v1.logging.set_verbosity(tf.compat.v1.logging.ERROR)
 
 class GPT2Generator:
 
-    def __init__(self,  generate_num=64, temperature=0.4, top_k=40, top_p=0.9):
+    def __init__(self,  generate_num=56, temperature=0.4, top_k=40, top_p=0.9):
         self.generate_num=generate_num
         self.temp = temperature
         self.top_k = top_k