Classifying images using Amazon SageMaker