Update README.md

README.md

@@ -19,15 +19,15 @@ We support the following languages/graphemaries:
 ```python
 from transformers import NllbTokenizerFast, AutoModelForSeq2SeqLM
 
-tokenizer = NllbTokenizerFast.from_pretrained("CenIA/nllb-200-3.3B-spa-arn", use_auth_token="HF_TOKEN")
-model = AutoModelForSeq2SeqLM.from_pretrained("CenIA/nllb-200-3.3B-spa-arn", use_auth_token="HF_TOKEN")
+tokenizer = NllbTokenizerFast.from_pretrained("CenIA/nllb-200-3.3B-spa-arn")
+model = AutoModelForSeq2SeqLM.from_pretrained("CenIA/nllb-200-3.3B-spa-arn")
 
 def translate(sentence: str, translate_from="spa_Latn", translate_to="quy_Latn") -> str:
     tokenizer.src_lang = translate_from
     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