LukeGPT88 commited on
Commit
ad7c036
1 Parent(s): 5fe3c89

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -102,13 +102,13 @@ It was fine-tuned on 14317 Reddit posts pulled from the (Reddit API) [https://pr
102
 
103
  ```python
104
  from transformers import pipeline
105
- classifier = pipeline("sentiment-analysis", model="michellejieli/NSFW_text_classification")
106
  classifier("I see you’ve set aside this special time to humiliate yourself in public.")
107
  ```
108
 
109
  ```python
110
  Output:
111
- [{'label': 'NSFW', 'score': 0.998853325843811}]
112
  ```
113
 
114
  # Contact
 
102
 
103
  ```python
104
  from transformers import pipeline
105
+ classifier = pipeline("text-classification", model="LukeGPT88/patient-doctor-text-classifier")
106
  classifier("I see you’ve set aside this special time to humiliate yourself in public.")
107
  ```
108
 
109
  ```python
110
  Output:
111
+ [{'label': 'NEUTRAL', 'score': 0.9890775680541992}]
112
  ```
113
 
114
  # Contact