Update app.py

app.py

@@ -6,11 +6,11 @@ model = BertModel.from_pretrained("bert-large-uncased")
 unmasker = pipeline('fill-mask', model=model, tokenizer=tokenizer)
 
 def fill_mask(sentence):
-    masked_sentence = sentence.replace("[MASK]", unmasker.tokenizer.mask_token)
-    results = unmasker(masked_sentence)
+    results = unmasker(sentence)
     return [result['sequence'] for result in results]
 
 inputs = gr.inputs.Textbox(label="Enter a sentence with a masked word")
 outputs = gr.outputs.Textbox(label="Predicted sequences")
 
 gr.Interface(fn=fill_mask, inputs=inputs, outputs=outputs).launch()
+