jayasuriyaK commited on
Commit
3935667
1 Parent(s): 2dfa08b

Update NSFW_text_classifier/README.md

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