Views
No views yet

1from transformers import pipeline
2text = "Hey little shit, GIVE ME YOUR SNACK !"
3classifier = pipeline("text-classification", model="oxyapi/albert-moderation-001", tokenizer="oxyapi/albert-moderation-001")
4result = classifier(text,top_k=len(classifier.model.config.id2label))
5print(result)
6@misc{albertmoderation0012025,
title={Albert Moderation 001: A fast and lightweight moderation model based on BERT},
author={Oxygen (oxyapi)},
year={2024},
howpublished={\url{https://huggingface.co/oxyapi/albert-moderation-001}},
}