Get insights by generating stories with AI
A text generator is an application based on NLP (Natural Language Processing) techniques. It tries to understand the context of an input and adds its own thoughts around it.
When provided a text input – a phrase, sentence or paragraph, this demo attempts to generate some stories around it. The demo was implemented using deep learning language models pre-trained on Wikipedia text.
Transfer learning was used to augment the pre-trained model on Yelp dataset.
The data set can be downloaded from here: https://s3.amazonaws.com/fast-ai-nlp/yelp_review_full_csv.tgz.
It contains 1,569,264 samples from the Yelp Dataset Challenge 2015
The demo is written using React for the user interface and Flask for implementing the REST API which interfaces with the language model.
The pre-trained model was downloaded from fast.ai