Error in running .py with tensorflow and keras while saving the results

Question

I have Keras 2.4.3 and TF 2.3.0 installed. I am using GPU for training my model. The training code is in ipynb format and py