Diffusion Model In Python Example Code For Beginners