Modelpublic

language-ml-lab/classification-azb

sourceHugging Faceupdated 3y agoView on Hugging Face

0likes9downloads

Model Card

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

python

# Use a pipeline as a high-level helper
from transformers import pipeline

pipe = pipeline("text-classification", model="language-ml-lab/classification-azb")

python

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