Creating the Model file using Scikit Learn