/GPT-2-EssayWriter

Essay writer that gathers training data from the one of two sites and trains it on a gpt-2 medium model.

Primary LanguagePythonMIT LicenseMIT

GPT-2-EssayWriter

GPT-2-EssayWriter uses the GPT-2 medium model created OpenAI. For simple access to the model I used HuggingFace. It contains a webscraper made in selenium to gather an infromation corpus for training

Usage

This repository is meant to be used as any particular text generation with GPT-2. The web scraper has two options. For literature it will go here for training data. For history or any subject that is not literature will go to wikipedia.

Contributing

Pull requests are welcome. For major changes, please open an issue first to discuss what you would like to change. Please make sure to update tests as appropriate.

License

MIT