It's the project which can be used to transform toxic sentences into friendlier ones.
To start my project you need to do several things:
- Download weights for the model. You need to put these unarchived weights into the models folder.
- Use this script to download data. Put the data folder
- Install requirements
pip3 install -r requirements.txt- Use train.py to train the model.
- Use test.py to test the model.
- Use predict_model.py to see how model works with your sentences.