fixed typos in README.md
Browse files
README.md
CHANGED
@@ -86,7 +86,7 @@ print(safe_label, categories)
|
|
86 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
87 |
import re
|
88 |
|
89 |
-
model_name = "Qwen/Qwen3Guard-
|
90 |
|
91 |
# load the tokenizer and the model
|
92 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
@@ -135,7 +135,7 @@ print(content)
|
|
135 |
# Refusal: Yes
|
136 |
# '''
|
137 |
safe_label, category_label, refusal_label = extract_label_categories_refusal(content)
|
138 |
-
print(safe_label,
|
139 |
```
|
140 |
|
141 |
## Deployment with SGLang and vLLM
|
|
|
86 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
87 |
import re
|
88 |
|
89 |
+
model_name = "Qwen/Qwen3Guard-Gen-0.6B"
|
90 |
|
91 |
# load the tokenizer and the model
|
92 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
|
|
135 |
# Refusal: Yes
|
136 |
# '''
|
137 |
safe_label, category_label, refusal_label = extract_label_categories_refusal(content)
|
138 |
+
print(safe_label, category_label, refusal_label)
|
139 |
```
|
140 |
|
141 |
## Deployment with SGLang and vLLM
|