Update app.py

app.py

@@ -4,7 +4,8 @@ from transformers import pipeline
 unmasker = pipeline('fill-mask', model='bert-large-uncased')
 
 def fill_mask(sentence):
-    results = unmasker(sentence)
+    masked_sentence = sentence.replace("[MASK]", unmasker.tokenizer.mask_token)
+    results = unmasker(masked_sentence)
     return [result['sequence'] for result in results]
 
 inputs = gr.inputs.Textbox(label="Enter a sentence with a masked word")