from diffusers import DDPMPipeline pipeline = DDPMPipeline.from_pretrained('ijiu/sd-class-butterflies-32') image = pipeline().images[0] image