Error while processing the HTML.ipynb file #69

I am trying to extract html from image files using the html.ipynb file but am stuck with issues in the below code:-
features = IR2.predict(images)
Default MaxPoolingOp only supports NHWC on device type CPU
[[{{node inception_resnet_v2/max_pooling2d/MaxPool}}]] [Op:__inference_predict_function_16619]

I am currently using Amazon sagemaker to train.
I have tried both tensorflow-gpu and tensorflow version 2.8.0 and keras 2.8.0

Any suggestions what could cause this issue