235.235. Masked Language Modeling

Implement a function to predict the next word in a sentence given a context. **Example:** Given the context 'I love to', the function should predict the next word. **Constraints:** The function should use a pre-trained language model and the input context should be a string.