Downloads · 30 days
24
8% of all-time downloads
Sekhinah/Talk_Shield_English
Talk_Shield_English is a machine learning model from Sekhinah. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
from transformers import AutoTokenizer, AutoModelForSequenceClassification
Downloads · 30 days
24
8% of all-time downloads
All-time downloads
290
Public
Parameters
125M
499 MB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors499 MB · 99%
From the Hugging Face model README
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("sekhinah/Talk_Shield_English") model = AutoModelForSequenceClassification.from_pretrained("sekhinah/Talk_Shield_English")
text = "You are such an idiot!" inputs = tokenizer(text, return_tensors="pt", truncation=True, padding=True) outputs = model(**inputs) prediction = outputs.logits.sigmoid() # multi-label classification print(prediction)