pt-sk commited on
Commit
5fd97b7
1 Parent(s): da99926

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -18,7 +18,7 @@ You can use the model with the following code.
18
  ```python
19
  from transformers import BertForSequenceClassification, BertTokenizer, TextClassificationPipeline
20
 
21
- model_path = "JungleLee/bert-toxic-comment-classification"
22
  tokenizer = BertTokenizer.from_pretrained(model_path)
23
  model = BertForSequenceClassification.from_pretrained(model_path, num_labels=2)
24
 
 
18
  ```python
19
  from transformers import BertForSequenceClassification, BertTokenizer, TextClassificationPipeline
20
 
21
+ model_path = "pt-sk/bert-toxic-classification"
22
  tokenizer = BertTokenizer.from_pretrained(model_path)
23
  model = BertForSequenceClassification.from_pretrained(model_path, num_labels=2)
24