How to run the project 1. Install dependencies pip install -r requirements.txt 2. Train the model python main.py --dataset <dataset directory>