Update README.md
Browse files
README.md
CHANGED
|
@@ -34,7 +34,7 @@ Classify Harmful messages and Detect PII via single forward pass
|
|
| 34 |
```python
|
| 35 |
from gliner2 import GLiNER2
|
| 36 |
|
| 37 |
-
model = GLiNER2.from_pretrained("
|
| 38 |
|
| 39 |
PII_LABELS = ["person", "address", "email", "phone"]
|
| 40 |
SAFETY_LABELS = ["safe", "unsafe"]
|
|
@@ -246,7 +246,7 @@ TOV_LABELS = [
|
|
| 246 |
]
|
| 247 |
|
| 248 |
# init model and schema
|
| 249 |
-
model = GLiNER2.from_pretrained("
|
| 250 |
schema = model.create_schema()
|
| 251 |
schema = schema.entities(entity_types=PII_LABELS, threshold=0.5)
|
| 252 |
schema = schema.classification(task="safety", labels=SAFETY_LABELS)
|
|
|
|
| 34 |
```python
|
| 35 |
from gliner2 import GLiNER2
|
| 36 |
|
| 37 |
+
model = GLiNER2.from_pretrained("hivetrace/gliner-guard-uniencoder")
|
| 38 |
|
| 39 |
PII_LABELS = ["person", "address", "email", "phone"]
|
| 40 |
SAFETY_LABELS = ["safe", "unsafe"]
|
|
|
|
| 246 |
]
|
| 247 |
|
| 248 |
# init model and schema
|
| 249 |
+
model = GLiNER2.from_pretrained("hivetrace/gliner-guard-uniencoder"
|
| 250 |
schema = model.create_schema()
|
| 251 |
schema = schema.entities(entity_types=PII_LABELS, threshold=0.5)
|
| 252 |
schema = schema.classification(task="safety", labels=SAFETY_LABELS)
|