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")