Diffusion Model Arxiv Api Python Code Example Input And Output