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
|