I am working on deep transfer learning. I train the model correctly and getting good accuracy. Nevertheless, I am facing trouble with the test set. I am facing an error in line: