This project uses a DistilBert model I fine tuned to perform binary classification on whether text is innapropriate or not. The model is then used in a discord bot to moderate messages sent in a discord server.
To use the model, first download the model files from This Google Drive Folder
Place the pytorch_model.bin and config.json files in the model folder.
Then, create a file called token.txt in the root directory of the project and paste your discord bot token in it.
Then run the following command to start the bot:
python3 bot.py
- transformers
- discord.py
- torch
- pandas