petaevd's picture
Publish multitask toxicity model
27c363f verified
Raw
History Blame Contribute Delete
261 Bytes
{
"architecture": "MultiTaskToxicityEncoder",
"base_model": "cointegrated/rubert-tiny2",
"labels": [
"profanity",
"threat",
"illegal"
],
"max_length": 176,
"thresholds": {
"profanity": 0.5,
"threat": 0.7,
"illegal": 0.9
}
}