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
- 8
Inference API (serverless) is not available, repository is disabled.