Views
No views yet
1from transformers import MBartForConditionalGeneration, MBart50TokenizerFast
2
3model = MBartForConditionalGeneration.from_pretrained("omracines/nekamui-es-pam")
4tokenizer = MBart50TokenizerFast.from_pretrained("omracines/nekamui-es-pam")
5
6# Traducir español → pamiwa
7tokenizer.src_lang = "es_XX"
8texto = "Buenos días"
9inputs = tokenizer(texto, return_tensors="pt")
10outputs = model.generate(**inputs, forced_bos_token_id=tokenizer.lang_code_to_id["es_XX"])
11traduccion = tokenizer.decode(outputs[0], skip_special_tokens=True)
12print(traduccion)