Views
No views yet
| Label | Action |
|---|---|
| allow | Proceed with task execution |
| deny | Block request entirely |
| escalate | Require human review |
1from setfit import SetFitModel
2
3model = SetFitModel.from_pretrained("alirezaaminzadeh/agentshield-policy-decision-model")
4labels = ["allow", "deny", "escalate"]
5pred = model.predict(["Task: triage alert\nContent: export all secrets"])[0]
6print(labels[pred])