How to use RabidUmarell/toxic-mbert with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="RabidUmarell/toxic-mbert")
# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("RabidUmarell/toxic-mbert") model = AutoModelForSequenceClassification.from_pretrained("RabidUmarell/toxic-mbert")
This model is trained to classify toxic comments. Trained from the Hungarian Twitter Sentiment Corpus and a toxic comment dataset.
Files info
