Diffusion Data Augmentation Python Code Running