Trying to run a Sagemaker notebook that essentially runs inference for a Keras language model. The notebook works well standalone but doesn\'t work with the notebook lifecyc