alirezaaminzadeh/agentshield-bench
Viewer • Updated • 5.76k • 26
How to use alirezaaminzadeh/agentshield-content-trust-classifier with setfit:
from setfit import SetFitModel
model = SetFitModel.from_pretrained("alirezaaminzadeh/agentshield-content-trust-classifier")Labels retrieval and untrusted content as trusted or untrusted for agent consumption.
from setfit import SetFitModel
model = SetFitModel.from_pretrained("alirezaaminzadeh/agentshield-content-trust-classifier")
print(model.predict(["Standard support ticket with no suspicious content"])) # [0]
Apache 2.0
Base model
nreimers/MiniLM-L6-H384-uncased