Testing the model on AWS SageMaker notebook