How to use martin-ha/toxic-comment-model with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="martin-ha/toxic-comment-model")
# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("martin-ha/toxic-comment-model") model = AutoModelForSequenceClassification.from_pretrained("martin-ha/toxic-comment-model")
· Sign up or log in to comment
