Train GPT-2 and generate text using custom datasets
Generate writing prompt based on selected categories