Using transformers to predict next word and predict <mask> word - View it on GitHub