ifmain
/

ModerationBERT-En-02

Text Classification

Model card Files Files and versions

ifmain commited on Apr 2

Commit

dcb5808

·

verified ·

1 Parent(s): 4bdda11

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -65,7 +65,7 @@ import torch
 from transformers import BertTokenizer, BertForSequenceClassification
 # Load the tokenizer and model
-model_name = "ifmain/ModerationBERT-ML-En"
 tokenizer = BertTokenizer.from_pretrained(model_name)
 model = BertForSequenceClassification.from_pretrained(model_name, num_labels=18)

 from transformers import BertTokenizer, BertForSequenceClassification
 # Load the tokenizer and model
+model_name = "ifmain/ModerationBERT-En-02"
 tokenizer = BertTokenizer.from_pretrained(model_name)
 model = BertForSequenceClassification.from_pretrained(model_name, num_labels=18)