Spaces:
Sleeping
Sleeping
Update tools/polite_guard.py
Browse files- tools/polite_guard.py +4 -2
tools/polite_guard.py
CHANGED
|
@@ -4,14 +4,16 @@ from transformers import pipeline
|
|
| 4 |
|
| 5 |
|
| 6 |
class PoliteGuardTool(Tool):
|
| 7 |
-
"""
|
| 8 |
-
|
|
|
|
| 9 |
|
| 10 |
Args:
|
| 11 |
input_text: Text that the user inputs into the agent and should then be evaluated.
|
| 12 |
|
| 13 |
Returns:
|
| 14 |
A classification label about whether the content is polite, somewhat polite, neutral or impolite.
|
|
|
|
| 15 |
"""
|
| 16 |
|
| 17 |
name = "polite_guard"
|
|
|
|
| 4 |
|
| 5 |
|
| 6 |
class PoliteGuardTool(Tool):
|
| 7 |
+
"""
|
| 8 |
+
Takes the input text from users and then evaluates it against Polite Guard to return specific information
|
| 9 |
+
about whether the content is polite.
|
| 10 |
|
| 11 |
Args:
|
| 12 |
input_text: Text that the user inputs into the agent and should then be evaluated.
|
| 13 |
|
| 14 |
Returns:
|
| 15 |
A classification label about whether the content is polite, somewhat polite, neutral or impolite.
|
| 16 |
+
|
| 17 |
"""
|
| 18 |
|
| 19 |
name = "polite_guard"
|