[Bug Report] I cannot send paramters to get predicted result from tensorflow_script_mode_training_and_serving.ipynb

https://github.com/aws/amazon-sagemaker-examples/blob/main/sagemaker-python-sdk/tensorflow_script_mode_training_and_serving/tensorflow_script_mode_training_and_serving.ipynb

I try to run that notebook. Then, I'm facing with getting predicted result from the notebook example.
I didn't change the code following the orignal notebook but why it still shows error as below.

The line of problem is
`predictions = predictor.predict(train_data[:50])`

Start with sagemaker-python-sdk/tensorflow_script_mode_training_and_serving/tensorflow_script_mode_training_and_serving.ipynb and run the notebook through the failing predictor.predict(train_data[:50]) call. Inspect the linked screenshot and the notebook's training and serving setup to identify why the input parameters are rejected. Done means the unchanged example produces a predicted result from that call.

Written by the indexing model from the issue text.