Views
No views yet
1
2from transformers import pipeline, AlbertTokenizer, AlbertForMaskedLM
3model = AlbertForMaskedLM.from_pretrained('josu/albert-pt-br')
4tokenizer = AlbertTokenizer.from_pretrained('josu/albert-pt-br')
5unmasker = pipeline('fill-mask', model=model, tokenizer=tokenizer ,device=0)
6
7text = 'Marte está no [MASK] solar.'
8unmasker(text)
9
10[{'score': 0.7004144191741943,
11 'token': 244,
12 'token_str': 'sistema',
13 'sequence': 'marte esta no sistema solar.'},
14 {'score': 0.02539917267858982,
15 'token': 4077,
16 'token_str': 'solar',
17 'sequence': 'marte esta no solar solar.'},
18 {'score': 0.020301498472690582,
19 'token': 49,
20 'token_str': 'seu',
21 'sequence': 'marte esta no seu solar.'},
22 {'score': 0.01753508299589157,
23 'token': 482,
24 'token_str': 'centro',
25 'sequence': 'marte esta no centro solar.'},
26 {'score': 0.013344300910830498,
27 'token': 1401,
28 'token_str': 'plano',
29 'sequence': 'marte esta no plano solar.'}]
30