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="yregelman/remix-command-classifier-dbrt")
# Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification

tokenizer = AutoTokenizer.from_pretrained("yregelman/remix-command-classifier-dbrt")
model = AutoModelForSequenceClassification.from_pretrained("yregelman/remix-command-classifier-dbrt")
Quick Links

Model Card for remix-command-classifier-dbrt

This model classifies text into the following categories:

  • new_creation
  • chat_message
  • nudge

Model Details

English only, based on distilbert-base-uncased model

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