add prediction for the validation set - 120 images