Hugging Face's logo

UngLong
/
cafebert-classification-ft

Text Classification
Transformers
Safetensors
Vietnamese
xlm-roberta
socialmedia
toxiccomment
classification_toxic_comment
transformer
text-embeddings-inference
Model card Files
xet
 Community
UngLong commited on
Commit
7cd3bca
·
verified ·
1 Parent(s): 579344a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -1
README.md CHANGED
@@ -15,6 +15,8 @@ tags:
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.
@@ -24,4 +26,4 @@ This model is a fine-tuned version of CafeBERT on a toxic comment classification
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")
 
15
  - socialmedia
16
  - toxiccomment
17
  - classification_toxic_comment
18
+ - transformer
19
+ library_name: transformers
20
  ---
21
  # CafeBERT Fine-tuned for Toxic Classification
22
  This model is a fine-tuned version of CafeBERT on a toxic comment classification task.
 
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")