Diffusion Models As Dataframe Python Add Column