Create command line options for model.py file - To train the model on a given dataset `python3 model.py --train path/to/dataset/` - To predict on a given image `python3 model.py --predict path/to/image/`