resolve a error

generate.py

@@ -11,7 +11,7 @@ def generate_text(model_data, input_text, max_new_token):
         model_pipeline = model_data["pipeline"]
         generated_text = model_pipeline(
             input_text,
-            max_length=max_new_token,
+            max_length=max_new_token + len(input_text.split()),  # افزایش max_length
             do_sample=False,  # غیرفعال کردن نمونه‌گیری (حالت حریصانه)
             truncation=True  # فعال کردن truncation
         )[0]["generated_text"]
@@ -40,7 +40,7 @@ def generate_text(model_data, input_text, max_new_token):
         outputs = model.generate(
             input_ids=input_ids,
             attention_mask=attention_mask,
-            max_new_tokens=max_new_token,
+            max_new_tokens=max_new_token,  # استفاده از max_new_tokens
             do_sample=False,  # غیرفعال کردن نمونه‌گیری (حالت حریصانه)
             pad_token_id=tokenizer.eos_token_id,
             repetition_penalty=1.2,