Update README.md

README.md

@@ -22,7 +22,7 @@ def translate(sentence: str, translate_from="spa_Latn", translate_to="mri_Latn")
     tokenizer.tgt_lang = translate_to
 
     inputs = tokenizer(sentence, return_tensors="pt")
-    result = model.generate(**inputs, forced_bos_token_id=tokenizer.lang_code_to_id[translate_to])
+    result = model.generate(**inputs, forced_bos_token_id=tokenizer.convert_tokens_to_ids(translate_to))
     decoded = tokenizer.batch_decode(result, skip_special_tokens=True)[0]
 
     return decoded