Update app.py

app.py

@@ -15,7 +15,7 @@ tokenizer = AutoTokenizer.from_pretrained('hackathon-pln-es/t5-small-spanish-nah
 
 def predict(input):
   input_ids = tokenizer('translate Spanish to Nahuatl: ' + input, return_tensors='pt').input_ids
-  outputs = model.generate(input_ids)
+  outputs = model.generate(input_ids, max_length=512)
   outputs = tokenizer.batch_decode(outputs, skip_special_tokens=True)[0]
   return outputs