Update README.md

README.md

@@ -33,7 +33,30 @@ These are controlnet weights trained on stabilityai/stable-diffusion-2-1-base wi
 #### How to use
 
 ```python
-# TODO: add an example code snippet for running this diffusion pipeline
+from PIL import Image
+from diffusers import (
+    ControlNetModel,
+    StableDiffusionControlNetPipeline,
+    UniPCMultistepScheduler,
+)
+
+checkpoint = "liuch37/controlnet-sd-2-1-base-v1"
+
+prompt = "YOUR_FAVORITE_PROMPT"
+
+control_image = Image.open("YOUR_SEMANTIC_IMAGE")
+
+controlnet = ControlNetModel.from_pretrained(checkpoint, torch_dtype=torch.float32)
+pipe = StableDiffusionControlNetPipeline.from_pretrained(
+    "stabilityai/stable-diffusion-2-1-base", controlnet=controlnet, torch_dtype=torch.float32
+)
+
+pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
+
+generator = torch.manual_seed(0)
+image = pipe(prompt, num_inference_steps=30, generator=generator, image=control_image).images[0]
+
+image.save("YOUR_OUTPUT_IMAGE")
 ```
 
 #### Limitations and bias