Update README.md

README.md

@@ -155,7 +155,21 @@ The model may exhibit biases based on the training data used. It may not perform
 
 Users should be aware of the limitations and biases of the model and use it accordingly. They should also be prepared to handle false positives and false negatives. It is recommended to fine-tune the model for specific downstream tasks and to evaluate its performance on relevant datasets.
 
-
+## Citation
+
+If you find our work useful, please consider citing us!
+
+```bibtex
+@misc{khader2025diffguardtextbasedsafetychecker,
+      title={DiffGuard: Text-Based Safety Checker for Diffusion Models}, 
+      author={Massine El Khader and Elias Al Bouzidi and Abdellah Oumida and Mohammed Sbaihi and Eliott Binard and Jean-Philippe Poli and Wassila Ouerdane and Boussad Addad and Katarzyna Kapusta},
+      year={2025},
+      eprint={2412.00064},
+      archivePrefix={arXiv},
+      primaryClass={cs.CV},
+      url={https://arxiv.org/abs/2412.00064}, 
+}
+```
 
 ### Load model directly
 ```python