Update app.py

app.py

@@ -17,7 +17,7 @@ tokenizer = MT5TokenizerFast.from_pretrained(
 
 def generate_text(input_text):
     input_ids = tokenizer.encode(input_text, return_tensors="pt")
-    output = model.generate(input_ids, max_new_tokens = len(input_text), num_beams = 1, early_stopping=True, length_penalty = -5.0)
+    output = model.generate(input_ids, max_new_tokens = len(input_text.split(' ')) * 2, num_beams = 2, early_stopping=True, length_penalty = -5.0)
     generated_text = tokenizer.decode(output[0], skip_special_tokens=True, clean_up_tokenization_spaces=True)
     return generated_text