Diffusion Model In Python Example Programming