I have a folder with test 3 files [img1.jpg, img2.jpg, img3.jpg] for prediction. I used class ImageFolder and Dataloader to serve my pretrained Pytorch model. I assume, ImageFo