File size: 261 Bytes
0cf0380
 
681036b
0cf0380
 
 
681036b
0cf0380
 
1
2
3
4
5
6
7
8
9
10
import transformers
from transformers import pipeline

unmasker = pipeline('fill-mask', model='bert-large-uncased')
user_input = input("Enter a sentence with a masked word: ")
results = unmasker(user_input)

for result in results:
    print(result['sequence'])