Setup

pip install -r requirements.txt

Train

python main.py