marzinouri commited on
Commit
b929b37
1 Parent(s): e5d9ea6

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +19 -0
README.md ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Text Classification Model
2
+ - Type: Fine-tuned BERT-based text classification model
3
+ - 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.
4
+
5
+ ## How to use
6
+ ```python
7
+ # Use a pipeline as a high-level helper
8
+ from transformers import pipeline
9
+
10
+ pipe = pipeline("text-classification", model="language-ml-lab/classification-azb")
11
+ ```
12
+
13
+ ```python
14
+ # Load model directly
15
+ from transformers import AutoTokenizer, AutoModelForSequenceClassification
16
+
17
+ tokenizer = AutoTokenizer.from_pretrained("language-ml-lab/classification-azb")
18
+ model = AutoModelForSequenceClassification.from_pretrained("language-ml-lab/classification-azb")
19
+ ```