Views
No views yet
1from transformers import pipeline
2
3classifier = pipeline("fill-mask", model="AI-Sweden-Models/bert-large-nordic-pile-1M-steps")
4classifier("Stockholm är Sveriges [MASK].")
5
6[{'score': 0.6612114906311035,
7 'token': 21168,
8 'token_str': 'huvudstad',
9 'sequence': 'Stockholm är Sveriges huvudstad.'},
10 {'score': 0.06771031022071838,
11 'token': 7625,
12 'token_str': 'hjärta',
13 'sequence': 'Stockholm är Sveriges hjärta.'},
14 {'score': 0.040731027722358704,
15 'token': 40158,
16 'token_str': 'framsida',
17 'sequence': 'Stockholm är Sveriges framsida.'},
18 {'score': 0.03236057981848717,
19 'token': 58649,
20 'token_str': 'mittpunkt',
21 'sequence': 'Stockholm är Sveriges mittpunkt.'},
22 {'score': 0.018861057236790657,
23 'token': 37202,
24 'token_str': 'navel',
25 'sequence': 'Stockholm är Sveriges navel.'}]