Update app.py

app.py

@@ -50,7 +50,7 @@ def translate(text, src_lang, tgt_lang):
         with torch.no_grad():
             translated_tokens = model.generate(
                 **inputs,
-                forced_bos_token_id=tokenizer.lang_code_to_id[tgt_code],
+                forced_bos_token_id=tokenizer.convert_tokens_to_ids(tgt_code),
                 max_length=512
             )