original readme

README.md

@@ -1,25 +1,4 @@
 ---
-license: apache-2.0
-base_model: distilbert-base-uncased
-model_name: Diffguard
-paper_link: https://arxiv.org/pdf/2412.00064
-pipeline_tag: text-classification
-tags:
-- Transformers
-- PyTorch
-- safety
-- inappropriate
-- distilbert
-- DiffGuard
-language:
-- en
-datasets:
-- eliasalbouzidi/NSFW-Safe-Dataset
-metrics:
-- f1
-- accuracy
-- precision
-- recall
 widget:
 - text: A family hiking in the mountains
   example_title: Safe
@@ -33,15 +12,33 @@ widget:
   example_title: Nsfw
 - text: A mass shooting
   example_title: Nsfw
+base_model: distilbert-base-uncased
+license: apache-2.0
+language:
+- en
+metrics:
+- f1
+- accuracy
+- precision
+- recall
+pipeline_tag: text-classification
+tags:
+- Transformers
+- ' PyTorch'
+- safety
+- innapropriate
+- distilbert
+datasets:
+- eliasalbouzidi/NSFW-Safe-Dataset
 model-index:
-- name: Diffguard
+- name: NSFW-Safe-Dataset
   results:
   - task:
       name: Text Classification
       type: text-classification
     dataset:
       name: NSFW-Safe-Dataset
-      type: eliasalbouzidi/NSFW-Safe-Dataset
+      type: .
     metrics:
     - name: F1
       type: f1
@@ -51,7 +48,7 @@ model-index:
       value: 0.98
 ---
 
-# DiffGuard Model Card
+# Model Card
 
 <!-- Provide a quick summary of what the model is/does. -->
 
@@ -74,10 +71,6 @@ 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.
@@ -180,5 +173,6 @@ from transformers import pipeline
 pipe = pipeline("text-classification", model="eliasalbouzidi/distilbert-nsfw-text-classifier")
 ```
 
+
 ## Contact
 Please reach out to [email protected] if you have any questions or feedback.