tdavidson/hate_speech_offensive
Viewer • Updated • 24.8k • 3.99k • 42
How to use Yash22CSU192/davidson-roberta-hatespeech with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="Yash22CSU192/davidson-roberta-hatespeech") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("Yash22CSU192/davidson-roberta-hatespeech")
model = AutoModelForSequenceClassification.from_pretrained("Yash22CSU192/davidson-roberta-hatespeech")from transformers import AutoModelForSequenceClassification, AutoTokenizer, pipeline
mid = "Yash22CSU192/davidson-roberta-hatespeech"
tok = AutoTokenizer.from_pretrained(mid) mdl = AutoModelForSequenceClassification.from_pretrained(mid)
clf = pipeline("text-classification", model=mdl, tokenizer=tok, return_all_scores=True)
print(clf("Have a nice day."))
Base model
FacebookAI/roberta-large