Commands to run 
1 - train.py
2 - docker build -t ml-model .
3 - docker run -p 5000:5000 ml-api