Text Classification Model
- Type: Fine-tuned BERT-based text classification model
- Description: This model has been fine-tuned using AzerBERT for text classification tasks. It is designed to categorize text into one of the following four categories: literature, sports, history, and geography.
How to use
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="language-ml-lab/classification-azb")
# Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("language-ml-lab/classification-azb")
model = AutoModelForSequenceClassification.from_pretrained("language-ml-lab/classification-azb")
- Downloads last month
- 186
Inference Providers
NEW
This model is not currently available via any of the supported third-party Inference Providers, and
the model is not deployed on the HF Inference API.