/GPT2-finetune

💭 Fine-tune a Covid-19 Doctor-like chatbot with GPT2

Primary LanguageJupyter NotebookMIT LicenseMIT

Finetune GPT2 on Covid-19 English Dataset

🙋🏻‍♀️ Create a Covid-19 Doctor-like chatbot with GPT2

🤷🏻‍♀️ Play with the demo on Colab or set it up yourself with the code!

Colab Demo

Open In Colab

To play with this yourself, you will need to have data-eng/ saved in your google drive, you can download them from this link.

💡 In the notebook, we will prepare the covid dataset for fine-tuning, train and evaluate the model, and get the conversation going with test data or interactive chat input from users!

Acknowledgement

Adapted from this very cool blog.