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).

Above is the dataset.