Train models to predict attributes of product images - View it on GitHub