Update app.py

app.py

@@ -33,7 +33,7 @@ def translate_text(input_text, sselected_language, tselected_language, model_nam
         prompt = f"translate {sselected_language} to {tselected_language}: {input_text}"
 
     input_ids = tokenizer.encode(prompt, return_tensors="pt")
-    output_ids = model.generate(input_ids)
+    output_ids = model.generate(input_ids, max_length=512)
     translated_text = tokenizer.decode(output_ids[0], skip_special_tokens=True)
 
     return translated_text