Diffusion Models As Dataframe Python Column Function In Python