My Final Project
I built an application to allow for generating news articles using OpenAI's GPT-2 text generator. The headline is provided by the user and they can optionally provide the start of the article, then it generates the rest of the article. The program has both a command line interface and a GUI, making it easy to use but also making it easier to integrate into other tools.
Link to Code
DanTm99 / gpt2-article-generator
An application that uses GPT-2 to generate news articles from user given prompts
GPT2 Article Generator
An application to allow for generating news articles using OpenAI's GPT-2 text generator. The model used for this was further trained on All The News, a dataset of over 200,000 news articles by components.one.
The repository can be cloned as normal:
git clone https://github.com/DanTm99/gpt2-article-generator.git
The model this program uses is hosted on Google Drive and can be downloaded from here. The contents of this archive should be extracted to the
gpt2-article-generator folder so that the
checkpoint is in the
Navigate into the folder:
To use this with your GPU you must have and NVIDIA GPU with a CUDA Compute Capability 3.5 or higher.
If you have the required hardware you must install the required software on your system as shown here.
Install the required packages as normal to use this with GPU support:
pip3 install -r…
How I built it (what's the stack? did I run into issues or discover something new along the way?)
I coded the project in Python and used TensorFlow as the models that were the focus on this project were TensorFlow models. In making this I learned how to interact with GPT-2, including further training it and using it to generate text in various ways.
Top comments (0)