Update README.md
Browse files
README.md
CHANGED
|
@@ -26,4 +26,6 @@ This model is a fine-tuned version of CafeBERT on a toxic comment classification
|
|
| 26 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
| 27 |
|
| 28 |
tokenizer = AutoTokenizer.from_pretrained("UngLong/cafebert-classification-ft")
|
| 29 |
-
model = AutoModelForSequenceClassification.from_pretrained("UngLong/cafebert-classification-ft")
|
|
|
|
|
|
|
|
|
| 26 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
| 27 |
|
| 28 |
tokenizer = AutoTokenizer.from_pretrained("UngLong/cafebert-classification-ft")
|
| 29 |
+
model = AutoModelForSequenceClassification.from_pretrained("UngLong/cafebert-classification-ft")
|
| 30 |
+
|
| 31 |
+

|