Pytorch model predicts same class for all images

Above link is my colab notebook for cnn model for mask recognition.Model gives the validation accuracy of 89 % and test accuracy of 85 % but when I try to predict the image by loading separately ,it gives me the same output for all images(with_mask).

https://drive.google.com/drive/folders/11EVQ15gH-0ODc6I0DR87sc9tVblkzEgg?usp=sharing

Above is the dataset.