from diffusers import DDPMPipeline pipeline = DDPMPipeline.from_pretrained('afshr/for_test') image = pipeline().images[0] image