How to use from the
Use from the
Transformers library
# Use a pipeline as a high-level helper
from transformers import pipeline

pipe = pipeline("text-classification", model="PedroRuizCode/ModernBERT-base-imdb-ft")
# Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification

tokenizer = AutoTokenizer.from_pretrained("PedroRuizCode/ModernBERT-base-imdb-ft")
model = AutoModelForSequenceClassification.from_pretrained("PedroRuizCode/ModernBERT-base-imdb-ft")
Quick Links

ModernBERT model trained for text classification using the IMDb dataset.

Just for fun

Downloads last month
1
Safetensors
Model size
0.1B params
Tensor type
F32
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for PedroRuizCode/ModernBERT-base-imdb-ft

Finetuned
(1260)
this model

Dataset used to train PedroRuizCode/ModernBERT-base-imdb-ft