I used the Keras example of the VAE as a base for my VAE implementation. For my use-case, I changed the layers and parameters accordingly to my images. After Training the re