Update README.md
Browse files
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
|
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(
|
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
|