How to use t-bank-ai/response-toxicity-classifier-base with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="t-bank-ai/response-toxicity-classifier-base") messages = [ {"role": "user", "content": "Who are you?"}, ] pipe(messages)
# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("t-bank-ai/response-toxicity-classifier-base") model = AutoModelForSequenceClassification.from_pretrained("t-bank-ai/response-toxicity-classifier-base")
Create images in seconds. No sign-up, no paywall, no setup.
SFconvertbot