Update README.md

README.md

@@ -1,3 +1,44 @@
----
-license: mit
----
+---
+license: mit
+---
+
+# Learning User Preferences for Image Generation Models
+
+
+[`Website`](https://learn-user-pref.github.io/) | [`Paper`](https://arxiv.org/abs/2508.08220) | [`GitHub`](https://github.com/Mowenyii/learn-user-pref) | [`BibTeX`](#citation)  
+
+
+
+
+
+## Installation
+For install instructions, please see https://github.com/Mowenyii/learn-user-pref.
+
+
+## Usage
+
+This model can be loaded from Hugging Face Hub as follows:
+
+```python
+from transformers import AutoProcessor, BitsAndBytesConfig, AutoModelForVision2Seq
+from peft import PeftModel
+model = AutoModelForVision2Seq.from_pretrained("HuggingFaceM4/idefics2-8b")
+model = PeftModel.from_pretrained(model, "wenyii/learn-user-pref")
+prompt_pool=torch.load("wenyii/learn-user-pref/prompt_pool.pt") 
+model.base_model.model.model.prompt_pool=prompt_pool
+```
+
+Please see https://github.com/Mowenyii/learn-user-pref for more detailed instructions.
+
+## Citation
+
+If you find this repository helpful, please consider citing our work:
+```
+@article{mo2025learning,
+  title={Learning User Preferences for Image Generation Model},
+  author={Mo, Wenyi and Ba, Ying and Zhang, Tianyu and Bai, Yalong and Li, Biye},
+  journal={arXiv preprint arXiv:2508.08220},
+  year={2025}
+}
+```
+