Generative Modeling Project 5 python code