Views
No views yet
1>>> from transformers import pipeline
2>>>
3>>> model = pipeline(
4>>> 'fill-mask',
5>>> model='rmihaylov/bert-base-theseus-bg',
6>>> tokenizer='rmihaylov/bert-base-theseus-bg',
7>>> device=0,
8>>> revision=None)
9>>> output = model("София е [MASK] на България.")
10>>> print(output)
11
12[{'score': 0.1586454212665558,
13 'sequence': 'София е столица на България.',
14 'token': 76074,
15 'token_str': 'столица'},
16 {'score': 0.12992817163467407,
17 'sequence': 'София е столица на България.',
18 'token': 2659,
19 'token_str': 'столица'},
20 {'score': 0.06064048036932945,
21 'sequence': 'София е Перлата на България.',
22 'token': 102146,
23 'token_str': 'Перлата'},
24 {'score': 0.034687548875808716,
25 'sequence': 'София е представителката на България.',
26 'token': 105456,
27 'token_str': 'представителката'},
28 {'score': 0.03053216263651848,
29 'sequence': 'София е присъединяването на България.',
30 'token': 18749,
31 'token_str': 'присъединяването'}]