Views
No views yet
facebook/mbart-large-50 base model.1from transformers import MBartForConditionalGeneration, MBart50TokenizerFast
2
3model_name = "rajeshrai577/roman-to-nepali-mbart50"
4
5model = MBartForConditionalGeneration.from_pretrained(model_name)
6tokenizer = MBart50TokenizerFast.from_pretrained(model_name, src_lang="en_XX", tgt_lang="ne_NP")
7
8input_text = "translate Roman Nepali to Nepali: ma ghar jaanchu."
9inputs = tokenizer(input_text, return_tensors="pt")
10generated_ids = model.generate(
11 **inputs,
12 forced_bos_token_id=tokenizer.lang_code_to_id["ne_NP"],
13 max_length=50
14)
15print(tokenizer.decode(generated_ids[0], skip_special_tokens=True))
16# Output: म घर जान्छु।