Update README.md

README.md

@@ -18,11 +18,11 @@ This model is a fine-tuned version of the [bert-base-uncased](https://huggingfac
 You can use the model with the following code.
 
 ```python
-from transformers import AutoModelForSequenceClassification, AutoTokenizer, TextClassificationPipeline
+from transformers import BertForSequenceClassification, BertTokenizer, TextClassificationPipeline
 
 model_path = "JungleLee/bert-toxic-comment-classification"
-tokenizer = AutoTokenizer.from_pretrained(model_path)
-model = AutoModelForSequenceClassification.from_pretrained(model_path)
+tokenizer = BertTokenizer.from_pretrained(model_path)
+model = BertForSequenceClassification.from_pretrained(model_path, num_labels=2)
 
 pipeline =  TextClassificationPipeline(model=model, tokenizer=tokenizer)
 print(pipeline("You're a fucking nerd."))