GAN implementation in Keras