responsible-ai commited on
Commit
85206d6
·
verified ·
1 Parent(s): bd66f9d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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 token: <SAFE> or <UNSAFE>.
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):