# Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("Alireza1044/bert_classification_lm")
model = AutoModelForSequenceClassification.from_pretrained("Alireza1044/bert_classification_lm")Quick Links
YAML Metadata Warning:empty or missing yaml metadata in repo card
Check out the documentation for more information.
A simple model trained on dialogues of characters in NBC series, The Office. The model can do a binary classification between Michael Scott and Dwight Shrute's dialogues.
| Label Definitions | |
|---|---|
| Label 0 | Michael |
| Label 1 | Dwight |
- Downloads last month
- 26
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="Alireza1044/bert_classification_lm")