Fine tune a GPT2 model on your WhatsApp and Telegram messages so that the bot responds to your messages as you would!
Screenshots:
| Let's get lunch | Let's hangout |
|---|---|
 |
 |
It's not perfect, but it definitely learned how to text like I would!
Download your messages from WhatsApp and Telegram
place the .txt files in the whatsapp_raw folder
place the result.json file in the telegram_raw folder
Follow the steps in the parseChats.ipynb notebook
Open trainModel.ipynb in Google Colab and upload all-messages.csv file
Download the fine-tuned model and place the directories inside this folder
Obtain a telegram bot token and paste it into the token.txt file
Verify that the path to the trained model in the bot_main.py file is correct
By this point you should have a telegram bot that answers questions as you would!
Feel free to make a pull request!
More Features are in the works!