Sorry for the late reply. Yeah, I used that repo to train the NN. Then I modified and reduced the code to only include the generative part, and that is what I added to the Kiosk-code.