distilbert-news-classifier

DistilBERT fine-tuned for multi-class news category classification on HuffPost dataset.

Usage

from transformers import pipeline
pipe = pipeline("text-classification", model="{HF_USERNAME}/distilbert-news-classifier")
print(pipe("AI transforms healthcare"))