Update README.md

README.md

@@ -17,7 +17,7 @@ widget:
 
 ---
 
-# Fine-tuned DistilRoBERTa-base for NSFW Classification
+# Fine-tuned DistilBERT for NSFW Inappropriate Text Classification
 
 # Model Description 
 
@@ -25,19 +25,19 @@ DistilBERT is a transformer model that performs sentiment analysis. I fine-tuned
 
 The model is a fine-tuned version of [DistilBERT](https://huggingface.co/docs/transformers/model_doc/distilbert).
 
-It was fine-tuned on 14317 Reddit posts pulled from the (Reddit API) [https://praw.readthedocs.io/en/stable/].
+It was fine-tuned on 19604 Reddit posts pulled from the (Comprehensive Abusiveness Detection Dataset) [https://aclanthology.org/2021.conll-1.43/].
 
 # How to Use 
 
 ```python
 from transformers import pipeline
-classifier = pipeline("sentiment-analysis", model="michellejieli/NSFW_text_classification")
+classifier = pipeline("sentiment-analysis", model="michellejieli/inappropriate_text_classifier")
 classifier("I see you’ve set aside this special time to humiliate yourself in public.")
 ```
 
 ```python
 Output:
-[{'label': 'NSFW', 'score': 0.998853325843811}]
+[{'label': 'NSFW', 'score': 0.9684491753578186}]
 ```
 
 # Contact