Spaces:
Sleeping
Sleeping
Update tools/polite_guard.py
Browse files- tools/polite_guard.py +1 -3
tools/polite_guard.py
CHANGED
|
@@ -30,9 +30,7 @@ class PoliteGuardTool(Tool):
|
|
| 30 |
input_text: The text to classify.
|
| 31 |
|
| 32 |
Returns:
|
| 33 |
-
|
| 34 |
-
a. A in result['label'] which is one of the four classifications
|
| 35 |
-
b. A score in result['score']
|
| 36 |
"""
|
| 37 |
try:
|
| 38 |
classifier = pipeline("text-classification", "Intel/polite-guard")
|
|
|
|
| 30 |
input_text: The text to classify.
|
| 31 |
|
| 32 |
Returns:
|
| 33 |
+
dictionary with label and score values
|
|
|
|
|
|
|
| 34 |
"""
|
| 35 |
try:
|
| 36 |
classifier = pipeline("text-classification", "Intel/polite-guard")
|