Views
No views yet
1from transformers import AutoTokenizer, AutoModelForMaskedLM, pipeline
2
3tokenizer = AutoTokenizer.from_pretrained("xlm-roberta-uganda-languages")
4model = AutoModelForMaskedLM.from_pretrained("xlm-roberta-uganda-languages")
5
6fill_mask = pipeline("fill-mask", model=model, tokenizer=tokenizer)
7result = fill_mask("Abantu b'omubyalo tibatera kwikiriza [MASK] muyaaka.")
8print(result)