Update README.md
Browse files
README.md
CHANGED
|
@@ -63,7 +63,7 @@ client = OpenAI(base_url="http://localhost:8000/v1", api_key="EMPTY") # vLLM de
|
|
| 63 |
|
| 64 |
SYSTEM_PROMPT = """
|
| 65 |
Task: Check if the LAST user message contains unsafe content.
|
| 66 |
-
Respond with only one
|
| 67 |
""".strip()
|
| 68 |
|
| 69 |
def classify(text: str, tau: float = 0.5):
|
|
|
|
| 63 |
|
| 64 |
SYSTEM_PROMPT = """
|
| 65 |
Task: Check if the LAST user message contains unsafe content.
|
| 66 |
+
Respond with only one word: safe or unsafe.
|
| 67 |
""".strip()
|
| 68 |
|
| 69 |
def classify(text: str, tau: float = 0.5):
|