Reflection removal

I already have a trained model. When loaded to perform reflection, it shows the screenshot error.
And, I ensure the input is to the specified dimension.

Attached is the error and the code.

Please any help to rectify and correct the error in handling the task.

Hi @Ashraf_Ishaq, If possible could you please share the stand alone code to reproduce the issue. It is difficult to understand from the error log.