/next-word-prediction-demo

A next-word-prediction demo written for Pencil Learning Technologies intern take home assignment.

Primary LanguageJupyter Notebook

Next Word Prediction Demo

Open in Colab

This demo predicts the next word of prompted half-sentences using Hugging Face's pre-trained OpenAI GPT-2 language model implemented in a Jupyter notebook. Feel free to click on the button above to run it in Google Colab.