model = MBartForConditionalGeneration.from_pretrained("facebook/mbart-large-50-many-to-many-mmt") | |
tokenizer = MBart50TokenizerFast.from_pretrained("facebook/mbart-large-50-many-to-many-mmt") | |
translate Hindi to French | |
tokenizer.src_lang = "hi_IN" | |
encoded_hi = tokenizer(article_hi, return_tensors="pt") | |
generated_tokens = model.generate(**encoded_hi, forced_bos_token_id=tokenizer.lang_code_to_id["fr_XX"]) | |
tokenizer.batch_decode(generated_tokens, skip_special_tokens=True) | |
=> "Le chef de l 'ONU affirme qu 'il n 'y a pas de solution militaire en Syria." |