Diffusion Models As Dataframe Python Loc