zenlm/zen-guard
1
0
Fork 0
mirror of https://github.com/zenlm/zen-guard.git synced 2026-07-27 06:13:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix print

Browse Source
This commit is contained in:
feihu.hf
2025-09-23 20:44:08 +08:00
parent fb4efb7031
commit e9aa28dab3
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
README.md
+2 -2
View File
@@ -298,9 +298,9 @@ for i in range(user_end_index + 1, len(token_ids)):
token_str = tokenizer.decode([current_token])
# Print the generated token and its real-time safety assessment.
if result['risk_level'][-1] == "Safe":
print(f"Token: '{token_str}' -> [Risk: {result['risk_level'][-1]}]")
print(f"Token: {repr(token_str)} -> [Risk: {result['risk_level'][-1]}]")
else:
print(f"Token: '{token_str}' -> [Risk: {result['risk_level'][-1]} - Category: {result['category'][-1]}]")
print(f"Token: {repr(token_str)} -> [Risk: {result['risk_level'][-1]} - Category: {result['category'][-1]}]")
```