Solving the issue "AttributeError: 'NllbTokenizerFast' object has no attribute 'lang_code_to_id'" (#1)

- Solving the issue "AttributeError: 'NllbTokenizerFast' object has no attribute 'lang_code_to_id'" (ffa8d951d25853987ad1da075ead6174d7fbd155)


Co-authored-by: Wendyellé Abubakrh Alban NYANTUDRE <[email protected]>

src/translation.py

@@ -94,7 +94,7 @@ def translate_facebook(s:str, src_iso:str, dest_iso:str)-> str:
 
     # Inference
     encoded = tokenizer(s, return_tensors="pt")
-    translated_tokens = model.generate(**encoded, forced_bos_token_id=tokenizer.lang_code_to_id[f"{dest_iso}_Latn"], max_length=30)
+    translated_tokens = model.generate(**encoded, forced_bos_token_id=tokenizer.convert_tokens_to_ids(f"{dest_iso}_Latn"), max_length=30)
     translation = tokenizer.batch_decode(translated_tokens, skip_special_tokens=True)[0]
 
     return translation