Update README.md

README.md

@@ -20,8 +20,33 @@ should probably proofread and complete it, then remove this comment. -->
 
 # ControlNet for cloth- Docty/cloth_controlnet
 
-These are LoRA adaption weights for stable-diffusion-v1-5/stable-diffusion-v1-5. The weights were trained on a realistic photo of poiuyt pipes using [DreamBooth](https://dreambooth.github.io/). You can find some example images in the following. 
+These are ControlNet for stable-diffusion-v1-5/stable-diffusion-v1-5.  You can find some example images in the following. 
 
 ![img_0](./image_control.png)
 ![img_1](./images_0.png)
-![img_2](./images_1.png)
+![img_2](./images_1.png)
+
+```python
+from diffusers import StableDiffusionControlNetPipeline, ControlNetModel, UniPCMultistepScheduler
+from diffusers.utils import load_image
+import torch
+
+base_model_path = "stable-diffusion-v1-5/stable-diffusion-v1-5"
+controlnet_path = "Docty/cloth_controlnet"
+
+controlnet = ControlNetModel.from_pretrained(controlnet_path, torch_dtype=torch.float16)
+pipe = StableDiffusionControlNetPipeline.from_pretrained(
+    base_model_path, controlnet=controlnet, torch_dtype=torch.float16
+)
+
+control_image = load_image("/kaggle/input/netcontroldataset/netcontrol/cond1.jpg")
+prompt = "a professional studio photograph of an attractive model wearing a teal top with lace detail"
+
+# generate image
+#generator = torch.manual_seed(0)
+image = pipe(
+    prompt, num_inference_steps=20, image=control_image
+).images[0]
+image
+
+```