Update app.py

app.py

@@ -83,9 +83,13 @@ def translate_text(text, src_lang, tgt_lang, models):
     
     for sentence in sentences:
         if sentence.strip():
-            # Removed src_lang from tokenizer call
             inputs = tokenizer(sentence, return_tensors="pt", padding=True, truncation=True, max_length=512)
-            translated = model.generate(**inputs, forced_bos_token_id=tokenizer.get_lang_id(tgt_lang_code), max_length=512)
+            # Use lang_code_to_id instead of get_lang_id
+            translated = model.generate(
+                **inputs,
+                forced_bos_token_id=tokenizer.lang_code_to_id[tgt_lang_code],
+                max_length=512
+            )
             translated_sentence = tokenizer.decode(translated[0], skip_special_tokens=True)
             translated_text += translated_sentence + "\n"