Hugging Face's logo

ifmain
/
ModerationBERT-ML-En

Text Classification
Transformers
Safetensors
English
bert
Inference Endpoints
Model card Files Community
1
ifmain commited on
Commit
a68f96b
·
verified ·
1 Parent(s): 724a72c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -63,7 +63,7 @@ import torch
63
  from transformers import BertTokenizer, BertForSequenceClassification
64
 
65
  # Load the tokenizer and model
66
- model_name = "ModerationBERT-ML-En"
67
  tokenizer = BertTokenizer.from_pretrained(model_name)
68
  model = BertForSequenceClassification.from_pretrained(model_name, num_labels=18)
69
 
 
63
  from transformers import BertTokenizer, BertForSequenceClassification
64
 
65
  # Load the tokenizer and model
66
+ model_name = "ifmain/ModerationBERT-ML-En"
67
  tokenizer = BertTokenizer.from_pretrained(model_name)
68
  model = BertForSequenceClassification.from_pretrained(model_name, num_labels=18)
69