[Fix] Update code for ML model

[ravi-prakash1907]

The following changes are requested on #59 to integrate the solution with UI:

1. Move the notebook (.ipynb) from the models to the code directory.
2. Using the provided jupyter notebook, save the final model (you may use pickle) and place it in models directory.

---

Convert both train and test data in a uniform format (either in Excel or CVS only).