for adding conditions to VAE I used this data set and labels but I got an error in part of the code:
here is my data generator:

112737×328 16.3 KB

and I got an error in this part:


114889×523 44.1 KB

how can I fix it?