UngLong
/

cafebert-classification-ft

Text Classification

classification_toxic_comment

text-embeddings-inference

Model card Files Files and versions

UngLong commited on Mar 20, 2025

Commit

7cd3bca

·

verified ·

1 Parent(s): 579344a

Update README.md

Files changed (1) hide show

README.md +3 -1

README.md CHANGED Viewed

@@ -15,6 +15,8 @@ tags:
 - socialmedia
 - toxiccomment
 - classification_toxic_comment
 ---
 # CafeBERT Fine-tuned for Toxic Classification
 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
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 tokenizer = AutoTokenizer.from_pretrained("UngLong/cafebert-classification-ft")
-model = AutoModelForSequenceClassification.from_pretrained("UngLong/cafebert-classification-ft")

 - socialmedia
 - toxiccomment
 - classification_toxic_comment
+- transformer
+library_name: transformers
 ---
 # CafeBERT Fine-tuned for Toxic Classification
 This model is a fine-tuned version of CafeBERT on a toxic comment classification task.
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 tokenizer = AutoTokenizer.from_pretrained("UngLong/cafebert-classification-ft")
+model = AutoModelForSequenceClassification.from_pretrained("UngLong/cafebert-classification-ft")