Update README.md

README.md

@@ -35,7 +35,7 @@ This `stable-diffusion-2-1-unclip` is a finetuned version of Stable Diffusion 2.
 
 ## Examples
 
-Using the [🤗's Diffusers library](https://github.com/huggingface/diffusers) to run Stable Diffusion UnCLIP 2-1 in a simple and efficient manner.
+Using the [🤗's Diffusers library](https://github.com/huggingface/diffusers) to run Stable Diffusion v2-1 UnCLIP in a simple and efficient manner.
 
 ```bash
 pip install git+https://github.com/diffusers.git transformers accelerate scipy safetensors
@@ -48,19 +48,24 @@ import torch
 from PIL import Image
 from io import BytesIO
 
-from diffusers import DiffusionPipeline
+from diffusers import StableUnCLIPImg2ImgPipeline
 
-pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-2-1-unclip", torch_dtype=torch.float16, variant="fp16")
-pipe.to("cuda")
+#Start the StableUnCLIP Image variations pipeline
+pipe = StableUnCLIPImg2ImgPipeline.from_pretrained(
+    "stabilityai/stable-diffusion-2-1-unclip", torch_dtype=torch.float16, variation="fp16"
+)
+pipe = pipe.to("cuda")
 
-# get image
+#Get image from URL
 url = "https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main/stable_unclip/tarsila_do_amaral.png"
 response = requests.get(url)
-image = Image.open(BytesIO(response.content)).convert("RGB")
+init_image = Image.open(BytesIO(response.content)).convert("RGB")
 
-# run image variation
-image = pipe(image).images[0]
+#Pipe to make the variation
+images = pipe(init_image).images
+images[0].save("tarsila_variation.png")
 ```
+Check out the [Stable UnCLIP pipeline docs here](https://huggingface.co/docs/diffusers/main/en/api/pipelines/stable_unclip)
 
 # Uses