Update app.py

app.py

@@ -13,7 +13,17 @@ translator = pipeline("translation", model=MODEL_PATH, token=TOKEN)
 # Translation function
 def translate_text(text):
     # Perform translation
-    translated = translator(text, max_length=512)
+    translated = translator(
+        text, 
+        max_length=512,
+        num_beams=4,  # Beam search for better quality
+        no_repeat_ngram_size=3,  # Avoid repetition
+        early_stopping=True,  # Stop when viable translations are found
+        do_sample=False,  # Deterministic output
+        pad_token_id=translator.tokenizer.pad_token_id,
+        bos_token_id=translator.tokenizer.bos_token_id,
+        eos_token_id=translator.tokenizer.eos_token_id
+    )
     return translated[0]["translation_text"]
 
 # Gradio app