Update README.md
Browse files
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("
|
106 |
classifier("I see you’ve set aside this special time to humiliate yourself in public.")
|
107 |
```
|
108 |
|
109 |
```python
|
110 |
Output:
|
111 |
-
[{'label': '
|
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
|