Update README.md

README.md

@@ -9,5 +9,34 @@ Our code and dataset are publicly available at https://prefpaint.github.io.
 # Usage
 
 ```bash
+import os
+from PIL import Image
+from diffusers import AutoPipelineForInpainting
 
+pipe = AutoPipelineForInpainting.from_pretrained(
+             'kd5678/prefpaint-v1.0', cache_dir='/data/kendong/cache')
+pipe = pipe.to("cuda")
+
+color_path = 'images.png'
+mask_path = 'mask.png'
+save_path = './results'
+os.makedirs(save_path, exist_ok=True)
+
+image, mask = Image.open(color_path), Image.open(mask_path).convert('L')
+# You can provide your prompt here.
+prompt = " "
+result = pipe(prompt=prompt, image=image, mask_image=mask, eta=1.0).images[0]            
+result.save(os.path.join(save_path, 'results.png'))
+  
+```
+
+# How to Cite
+
+```
+@article{liu2024prefpaint,
+  title={PrefPaint: Aligning Image Inpainting Diffusion Model with Human Preference},
+  author={Liu, Kendong and Zhu, Zhiyu and Li, Chuanhao and Liu, Hui and Zeng, Huanqiang and Hou, Junhui},
+  journal={arXiv preprint arXiv:2410.21966},
+  year={2024}
+}
 ```