Update README.md

README.md

@@ -25,9 +25,9 @@ from PIL import Image
 from diffusers import DiffusionPipeline
 
 pipeline = DiffusionPipeline.from_pretrained("superdiff/superdiff-sd-v1-4", custom_pipeline='pipeline', trust_remote_code=True)
-image = pipeline("a flamingo", "a candy cane", seed=1, num_inference_steps=1000, batch_size=1)
+output = pipeline("a flamingo", "a candy cane", seed=1, num_inference_steps=1000, batch_size=1)
 
-image = Image.fromarray(image.cpu().numpy())
+image = Image.fromarray(output[0].cpu().numpy())
 image.save("superdiff_output.png")
 ```