Diffusion Based Data Augmentation Python