Update README.md

README.md

@@ -1,3 +1,35 @@
----
-license: mit
----
+---
+library_name: diffusers
+license: mit
+pipeline_tag: text-to-image
+base_model:
+- black-forest-labs/FLUX.1-dev
+---
+
+# Model Summary
+This model is trained using [UnifiedReward-Flex](https://huggingface.co/collections/CodeGoat24/unifiedreward-flex) as reward on the training dataset of [UniGenBench](https://github.com/CodeGoat24/UniGenBench).
+
+🚀 The inference code is available at [Github](https://github.com/CodeGoat24/Pref-GRPO/blob/main/inference/flux_dist_infer.sh).
+
+
+For further details, please refer to the following resources:
+- 📰 Paper: https://arxiv.org/abs/2602.02380
+- 🪐 Project Page: https://codegoat24.github.io/UnifiedReward/flex
+- 🤗 Model Collections: https://huggingface.co/collections/CodeGoat24/unifiedreward-flex
+- 🤗 Dataset: https://huggingface.co/datasets/CodeGoat24/UnifiedReward-Flex-SFT-90K
+- 👋 Point of Contact: [Yibin Wang](https://codegoat24.github.io)
+
+
+
+
+
+## Citation
+
+```bibtex
+@article{unifiedreward-flex,
+  title={Unified Personalized Reward Model for Vision Generation},
+  author={Wang, Yibin and Zang, Yuhang and Han, Feng and Zhou, Yujie and Bu, Jiazi and Jin, Cheng and Wang, Jiaqi},
+  journal={arXiv preprint arXiv:2602.02380},
+  year={2026}
+}
+```