Add model name, paper link and citation instructions (#2)
Browse files- Add model name, paper link and citation instructions (e7d34ec0444962731ea65621405b2f8afc842f42)
Co-authored-by: Abdellah OUMIDA <Meeex2@users.noreply.huggingface.co>
README.md
CHANGED
|
@@ -28,6 +28,7 @@ tags:
|
|
| 28 |
- safety
|
| 29 |
- innapropriate
|
| 30 |
- distilbert
|
|
|
|
| 31 |
datasets:
|
| 32 |
- eliasalbouzidi/NSFW-Safe-Dataset
|
| 33 |
model-index:
|
|
@@ -71,6 +72,10 @@ The model can be used directly to classify text into one of the two classes. It
|
|
| 71 |
- **Language (NLP):** English
|
| 72 |
- **License:** apache-2.0
|
| 73 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 74 |
### Uses
|
| 75 |
|
| 76 |
The model can be integrated into larger systems for content moderation or filtering.
|
|
@@ -173,6 +178,22 @@ from transformers import pipeline
|
|
| 173 |
pipe = pipeline("text-classification", model="eliasalbouzidi/distilbert-nsfw-text-classifier")
|
| 174 |
```
|
| 175 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 176 |
|
| 177 |
## Contact
|
| 178 |
Please reach out to eliasalbouzidi@gmail.com if you have any questions or feedback.
|
|
|
|
| 28 |
- safety
|
| 29 |
- innapropriate
|
| 30 |
- distilbert
|
| 31 |
+
- DiffGuard
|
| 32 |
datasets:
|
| 33 |
- eliasalbouzidi/NSFW-Safe-Dataset
|
| 34 |
model-index:
|
|
|
|
| 72 |
- **Language (NLP):** English
|
| 73 |
- **License:** apache-2.0
|
| 74 |
|
| 75 |
+
## Technical Paper:
|
| 76 |
+
|
| 77 |
+
A more detailed technical overview of the model and the dataset can be found [here](https://arxiv.org/pdf/2412.00064).
|
| 78 |
+
|
| 79 |
### Uses
|
| 80 |
|
| 81 |
The model can be integrated into larger systems for content moderation or filtering.
|
|
|
|
| 178 |
pipe = pipeline("text-classification", model="eliasalbouzidi/distilbert-nsfw-text-classifier")
|
| 179 |
```
|
| 180 |
|
| 181 |
+
## Citation
|
| 182 |
+
|
| 183 |
+
If you find our work useful, please consider citing us!
|
| 184 |
+
|
| 185 |
+
```bibtex
|
| 186 |
+
@misc{khader2025diffguardtextbasedsafetychecker,
|
| 187 |
+
title={DiffGuard: Text-Based Safety Checker for Diffusion Models},
|
| 188 |
+
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},
|
| 189 |
+
year={2025},
|
| 190 |
+
eprint={2412.00064},
|
| 191 |
+
archivePrefix={arXiv},
|
| 192 |
+
primaryClass={cs.CV},
|
| 193 |
+
url={https://arxiv.org/abs/2412.00064},
|
| 194 |
+
}
|
| 195 |
+
```
|
| 196 |
+
|
| 197 |
|
| 198 |
## Contact
|
| 199 |
Please reach out to eliasalbouzidi@gmail.com if you have any questions or feedback.
|