Image Modeling Machine Learning Python Code