Flask API for training and predicting using scikit learn models - View it on GitHub