Using transformers to predict next word and predict <mask> word
Using transformers to predict next word and predict <mask> word
Simple application using transformers models to predict next word or a masked word in a sentence.
The purpose is to demo and compare the main models available up to date.
The first load take a long time since the application will download all the models. Beside 6 models running, inference time is acceptable even in CPU.
This app implements two variants of the same task (predict token). The first one consider the is at end of the sentence, simulating a prediction of the next word of the sentece.
The second variant is necessary to include a token where you want the model to predict the word.
cd web-app
python app.pyOpen your browser http://localhost:8000
No open issues yet, or sync has not completed.