I have used IMDB movie review dataset to create custom model by using DistilBertForSequenceClassification.

from transformers import DistilBertForSequenceClassification, Trainer, TrainingArguments

model = DistilBertForSequenceClassification.from_pretrained('./imdbsentdistilbertmodel')

Inference Providers NEW

This model isn't deployed by any Inference Provider. 🙋 Ask for provider support