Base classes for deployment of ML models - View it on GitHub