Download and extract the following datasets in a new folder called datasets:
Run create_dataset.ipynb to create train.csv and valid.csv
create_dataset.ipynb
Run train.ipynb to train the model
train.ipynb