ifmain
/

ModerationBERT-ML-En

Text Classification

Inference Endpoints

Model card Files Files and versions Community

ifmain commited on May 24

Commit

a68f96b

•

1 Parent(s): 724a72c

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -63,7 +63,7 @@ import torch
 from transformers import BertTokenizer, BertForSequenceClassification
 # Load the tokenizer and model
-model_name = "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-ML-En"
 tokenizer = BertTokenizer.from_pretrained(model_name)
 model = BertForSequenceClassification.from_pretrained(model_name, num_labels=18)