thon | |
from transformers import MBartForConditionalGeneration, MBartTokenizer | |
tokenizer = MBartTokenizer.from_pretrained("facebook/mbart-large-en-ro", src_lang="en_XX") | |
article = "UN Chief Says There Is No Military Solution in Syria" | |
inputs = tokenizer(article, return_tensors="pt") | |
translated_tokens = model.generate(**inputs, decoder_start_token_id=tokenizer.lang_code_to_id["ro_RO"]) | |
tokenizer.batch_decode(translated_tokens, skip_special_tokens=True)[0] | |
"Şeful ONU declară că nu există o soluţie militară în Siria" | |
Overview of MBart-50 | |
MBart-50 was introduced in the Multilingual Translation with Extensible Multilingual Pretraining and Finetuning paper by Yuqing Tang, Chau Tran, Xian Li, Peng-Jen Chen, Naman Goyal, Vishrav | |
Chaudhary, Jiatao Gu, Angela Fan. |