Text Classification
Transformers
TensorBoard
Safetensors
English
modernbert
text-embeddings-inference
Instructions to use PedroRuizCode/ModernBERT-base-imdb-ft with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use PedroRuizCode/ModernBERT-base-imdb-ft with Transformers:
# 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") - Notebooks
- Google Colab
- Kaggle
Update README.md
Browse files
README.md
CHANGED
|
@@ -6,6 +6,7 @@ base_model:
|
|
| 6 |
pipeline_tag: text-classification
|
| 7 |
datasets:
|
| 8 |
- stanfordnlp/imdb
|
|
|
|
| 9 |
---
|
| 10 |
|
| 11 |
ModernBERT model trained for text classification using the IMDb dataset.
|
|
|
|
| 6 |
pipeline_tag: text-classification
|
| 7 |
datasets:
|
| 8 |
- stanfordnlp/imdb
|
| 9 |
+
library_name: transformers
|
| 10 |
---
|
| 11 |
|
| 12 |
ModernBERT model trained for text classification using the IMDb dataset.
|