Python Keras Convolutional autoencoder. ValueError: logits and labels must have the same shape

Question

I took most of this code from here. I am very new to autoencoders.

Heres the code:

input_img = keras.Input(shape=(64, 55,1))

x = keras.layers.Conv2D(16, (