Update README.md

README.md

@@ -17,5 +17,5 @@ mask_image = load_image("https://huggingface.co/datasets/huggingface/documentati
 generator = torch.Generator("cuda").manual_seed(92)
 prompt = "concept art digital painting of an elven castle, inspired by lord of the rings, highly detailed, 8k"
 image = pipeline(prompt=prompt, image=init_image, mask_image=mask_image, generator=generator).images[0]
-make_image_grid([init_image, image], rows=1, cols=2)
+make_image_grid([init_image, mask_image, image], rows=1, cols=3)
 ```