Generative Modeling Project 5 oython code