Update app.py

app.py

@@ -19,9 +19,8 @@ language_codes = {
 def translate_text(text, target_lang_code):
     # Prepare the input and translate
     inputs = tokenizer.encode(text, return_tensors="pt")
-    # Set the decoder start token ID based on language code
-    decoder_start_token_id = tokenizer.convert_tokens_to_ids(f"<{target_lang_code}>")
-    translated = model.generate(inputs, decoder_start_token_id=decoder_start_token_id)
+    # Generate the translation
+    translated = model.generate(inputs, decoder_start_token_id=tokenizer.get_lang_id(target_lang_code))
     translated_text = tokenizer.decode(translated[0], skip_special_tokens=True)
     return translated_text