from diffusers import StableDiffusionPipeline pipeline = StableDiffusionPipeline.from_pretrained('dogeplusplus/stable-sam') image = pipeline().images[0] image