Update README.md
Browse files
README.md
CHANGED
|
@@ -38,6 +38,8 @@ This is the model card of a 🤗 transformers model that has been pushed on the
|
|
| 38 |
|
| 39 |
```bash
|
| 40 |
from tqdm import tqdm
|
|
|
|
|
|
|
| 41 |
model = AutoModelForSequenceClassification.from_pretrained(
|
| 42 |
'Fidlobabovic/censor-v5.9.1-preview', token='токен',
|
| 43 |
num_labels=num_labels,
|
|
@@ -45,6 +47,8 @@ model = AutoModelForSequenceClassification.from_pretrained(
|
|
| 45 |
label2id=label2id,
|
| 46 |
ignore_mismatched_sizes=True
|
| 47 |
)
|
|
|
|
|
|
|
| 48 |
|
| 49 |
def classify_questions_batch(texts, batch_size=32):
|
| 50 |
"""
|
|
|
|
| 38 |
|
| 39 |
```bash
|
| 40 |
from tqdm import tqdm
|
| 41 |
+
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
| 42 |
+
|
| 43 |
model = AutoModelForSequenceClassification.from_pretrained(
|
| 44 |
'Fidlobabovic/censor-v5.9.1-preview', token='токен',
|
| 45 |
num_labels=num_labels,
|
|
|
|
| 47 |
label2id=label2id,
|
| 48 |
ignore_mismatched_sizes=True
|
| 49 |
)
|
| 50 |
+
tokenizer = AutoTokenizer.from_pretrained('Fidlobabovic/censor-v5.9.1-preview')
|
| 51 |
+
|
| 52 |
|
| 53 |
def classify_questions_batch(texts, batch_size=32):
|
| 54 |
"""
|