michellejieli commited on
Commit
ee7026f
1 Parent(s): 73ed716

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +46 -0
README.md CHANGED
@@ -1,3 +1,49 @@
1
  ---
2
  license: creativeml-openrail-m
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: creativeml-openrail-m
3
  ---
4
+ ---
5
+ language: "en"
6
+ tags:
7
+ - distilroberta
8
+ - sentiment
9
+ - NSFW
10
+ - inappropriate
11
+ - spam
12
+ - twitter
13
+ - reddit
14
+
15
+ widget:
16
+ - text: "I like you. You remind me of me when I was young and stupid."
17
+ - text: "I see you’ve set aside this special time to humiliate yourself in public."
18
+ - text: "Have a great weekend! See you next week!"
19
+
20
+ ---
21
+
22
+ # Fine-tuned DistilRoBERTa-base for NSFW Classification
23
+
24
+ # Model Description
25
+
26
+ DistilBERT is a transformer model that performs sentiment analysis. I fine-tuned the model on Reddit posts with the purpose of classifying not safe for work (NSFW) content, specifically text that is considered inappropriate and unprofessional. The model predicts 2 classes, which are NSFW or safe for work (SFW).
27
+
28
+ The model is a fine-tuned version of [DistilBERT](https://huggingface.co/docs/transformers/model_doc/distilbert).
29
+
30
+ It was fine-tuned on 14317 Reddit posts pulled from the (Reddit API) [https://praw.readthedocs.io/en/stable/].
31
+
32
+ # How to Use
33
+
34
+ ```python
35
+ from transformers import pipeline
36
+ classifier = pipeline("sentiment-analysis", model="michellejieli/NSFW_text_classification")
37
+ classifier("I see you’ve set aside this special time to humiliate yourself in public.")
38
+ ```
39
+
40
+ ```python
41
+ Output:
42
+ [{'label': 'NSFW', 'score': 0.998853325843811}]
43
+ ```
44
+
45
+ # Contact
46
+
47
+ Please reach out to [michelle.li851@duke.edu](mailto:michelle.li851@duke.edu) if you have any questions or feedback.
48
+
49
+ ---