GitHub repository: trisongz/aitextgen
Description: A robust Python tool for text-based AI training and generation using GPT-2.