Update src/translation.py

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=120)
+    translated_tokens = model.generate(**encoded, forced_bos_token_id=tokenizer.convert_tokens_to_ids[f"{dest_iso}_Latn"], max_length=120)
     translation = tokenizer.batch_decode(translated_tokens, skip_special_tokens=True)[0]
 
     return translation