Views
No views yet
1from transformers import AutoModelForMaskedLM, AutoTokenizer, pipeline
2
3model_id = "proxectonos/deberta-xsmall-gl"
4
5tokenizer = AutoTokenizer.from_pretrained(model_id)
6model = AutoModelForMaskedLM.from_pretrained(model_id)
7
8fill_mask = pipeline("fill-mask", model=model, tokenizer=tokenizer)
9
10fill_mask("O Parlamento de Galicia aprobou a <mask> hoxe.")