muneebashraf's picture
Update app.py
0cf0380
raw
history blame
261 Bytes
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'])