Image classification validating results

I am excited to say I completed my first CNN image classification model using a dataset I created from scratch. I wasn’t expecting to be able to complete something like this so quickly but the course really really does make learning a complex subject matter easy. I am hoping I can have some assistance creating a validation check after the model has run that confirms the classification is correct. I know I have the results from the model but I would like to visualise each image either by correlating the classification to the image by the images file name or visually. I am currently using a yhat method but need to supply the file name each time i run the check if a mentor could dm me or I can share the code here (not related to course assessments I used a combination of different models) I would appreciate it, I cannot share the notebook or dataset