Text Classification
Transformers
PyTorch
English
distilbert
classification
sequence-classification
text-embeddings-inference
# Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("profoz/mlops-demo")
model = AutoModelForSequenceClassification.from_pretrained("profoz/mlops-demo")Quick Links
Github repository here
- Downloads last month
- 7
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="profoz/mlops-demo")