Here's an example of that with a smaller BERT model: https://pair.withgoogle.com/explorables/fill-in-the-blank/