Add model name, paper link and citation instructions (#2)

- Add model name, paper link and citation instructions (e7d34ec0444962731ea65621405b2f8afc842f42)


Co-authored-by: Abdellah OUMIDA <[email protected]>

README.md

@@ -28,6 +28,7 @@ tags:
 - safety
 - innapropriate
 - distilbert
+- DiffGuard
 datasets:
 - eliasalbouzidi/NSFW-Safe-Dataset
 model-index:
@@ -71,6 +72,10 @@ The model can be used directly to classify text into one of the two classes. It
 - **Language (NLP):** English
 - **License:** apache-2.0
 
+## Technical Paper:
+
+A more detailed technical overview of the model and the dataset can be found [here](https://arxiv.org/pdf/2412.00064).
+
 ### Uses
 
 The model can be integrated into larger systems for content moderation or filtering.
@@ -173,6 +178,22 @@ from transformers import pipeline
 pipe = pipeline("text-classification", model="eliasalbouzidi/distilbert-nsfw-text-classifier")
 ```
 
+## 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}, 
+}
+```
+
 
 ## Contact
 Please reach out to [email protected] if you have any questions or feedback.