| # Text Classification Model | |
| - Type: Fine-tuned BERT-based text classification model | |
| - Description: This model has been fine-tuned using [AzerBERT](https://huggingface.co/language-ml-lab/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") | |
| ``` |