Applying CGANs on more than 2 classes

[Zardian18]

Following the book, ive been trying to apply ConditionalWGAN for fashion mnist dataset with 10 classes but am facing error. I changed the model accordingly for both generator and critic and instead of one hot encoding the labels, im passing them just as numbers. What are the changes I am supposed to make. I've tried a few but none works.
And if i try to fit it normally, this error occurs. Does anyone know how to implement it?