Update README.md

README.md

@@ -18,16 +18,30 @@ library_name: diffusers
 
 ## Diffusers
 ```python
+from io import BytesIO
+from PIL import Image
+import requests
 from diffusers import AsymmetricAutoencoderKL, StableDiffusionInpaintPipeline
 
-pipe = StableDiffusionInpaintPipeline.from_pretrained(
-    "runwayml/stable-diffusion-inpainting"
-)
-pipe.vae = AsymmetricAutoencoderKL.from_pretrained(
-    "cross-attention/asymmetric-autoencoder-kl-x-2"
-)
+
+def download_image(url: str) -> Image.Image:
+    response = requests.get(url)
+    return Image.open(BytesIO(response.content)).convert("RGB")
+
+
+prompt = "a photo of a person"
+img_url = "https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main/repaint/celeba_hq_256.png"
+mask_url = "https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main/repaint/mask_256.png"
+
+image = download_image(img_url).resize((256, 256))
+mask_image = download_image(mask_url).resize((256, 256))
+
+pipe = StableDiffusionInpaintPipeline.from_pretrained("runwayml/stable-diffusion-inpainting")
+pipe.vae = AsymmetricAutoencoderKL.from_pretrained("cross-attention/asymmetric-autoencoder-kl-x-2")
+pipe.to("cuda")
 
 image = pipe(prompt=prompt, image=image, mask_image=mask_image).images[0]
+image.save("image.jpeg")
 ```
 
 ### Visual