Training and serving the TensorFlow model through SageMaker