Update README.md

README.md

@@ -2,8 +2,29 @@
 tags:
 - model_hub_mixin
 - pytorch_model_hub_mixin
+datasets:
+- atfortes/BokehDiffusion
+base_model:
+- black-forest-labs/FLUX.1-dev
+pipeline_tag: text-to-image
+library_name: diffusers
 ---
 
-This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration:
-- Library: [More Information Needed]
-- Docs: [More Information Needed]
+# Bokeh Diffusion
+
+This is the model checkpoint of ["Bokeh Diffusion: Defocus Blur Control in Text-to-Image Diffusion Models"](https://atfortes.github.io/projects/bokeh-diffusion/) (ACM SIGGRAPH Asia 2025).
+- Project: [atfortes.github.io/projects/bokeh-diffusion/](https://atfortes.github.io/projects/bokeh-diffusion/)
+- Paper: [arxiv/2503.08434](https://arxiv.org/abs/2503.08434)
+- GitHub: [atfortes/BokehDiffusion](https://github.com/atfortes/BokehDiffusion)
+
+## Citation
+
+If you use this dataset, please cite our work:
+```bibtex
+@article{fortes2025bokeh,
+    title   = {Bokeh Diffusion: Defocus Blur Control in Text-to-Image Diffusion Models},
+    author  = {Fortes, Armando and Wei, Tianyi and Zhou, Shangchen and Pan, Xingang},
+    journal = {arXiv preprint arXiv:2503.08434},
+    year    = {2025},
+}
+```