Try adding these lines:

x_train = np.expand_dims(x_train.astype(‘float32’) / 255, 3)

x_test = np.expand_dims(x_test.astype(‘float32’) / 255, 3)

y_train = np.expand_dims(y_train, 1)

y_test = np.expand_dims(y_test, 1)