Error while processing the HTML.ipynb file

[kingafy]

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