UngLong commited on
Commit
579344a
·
verified ·
1 Parent(s): f6104c8

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +10 -1
README.md CHANGED
@@ -15,4 +15,13 @@ tags:
15
  - socialmedia
16
  - toxiccomment
17
  - classification_toxic_comment
18
- ---
 
 
 
 
 
 
 
 
 
 
15
  - socialmedia
16
  - toxiccomment
17
  - classification_toxic_comment
18
+ ---
19
+ # CafeBERT Fine-tuned for Toxic Classification
20
+ This model is a fine-tuned version of CafeBERT on a toxic comment classification task.
21
+
22
+ ## Usage
23
+ ```python
24
+ from transformers import AutoTokenizer, AutoModelForSequenceClassification
25
+
26
+ tokenizer = AutoTokenizer.from_pretrained("UngLong/cafebert-classification-ft")
27
+ model = AutoModelForSequenceClassification.from_pretrained("UngLong/cafebert-classification-ft")