JungleLee commited on
Commit
bac46c7
1 Parent(s): c5769ef

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -11,7 +11,7 @@ pipeline_tag: text-classification
11
  ---
12
 
13
  ## Model description
14
- This model is a fine-tuned version of the [bert-base-uncased model](https://huggingface.co/transformers/model_doc/bert.html) to classify toxic comments.
15
 
16
  ## How to use
17
 
@@ -25,7 +25,7 @@ tokenizer = AutoTokenizer.from_pretrained(model_path)
25
  model = AutoModelForSequenceClassification.from_pretrained(model_path)
26
 
27
  pipeline = TextClassificationPipeline(model=model, tokenizer=tokenizer)
28
- print(pipeline('You're a fucking nerd.'))
29
  ```
30
 
31
  ## Training data
 
11
  ---
12
 
13
  ## Model description
14
+ This model is a fine-tuned version of the [bert-base-uncased](https://huggingface.co/transformers/model_doc/bert.html) model to classify toxic comments.
15
 
16
  ## How to use
17
 
 
25
  model = AutoModelForSequenceClassification.from_pretrained(model_path)
26
 
27
  pipeline = TextClassificationPipeline(model=model, tokenizer=tokenizer)
28
+ print(pipeline("You're a fucking nerd."))
29
  ```
30
 
31
  ## Training data