From 9a0a13b5c8a9ee223bb4ee6c8aef4fc6c9541c83 Mon Sep 17 00:00:00 2001
From: Mu Yang <emfomy@gmail.com>
Date: Wed, 6 Jan 2021 13:47:42 +0800
Subject: [PATCH] Update usage.

---
 README.md | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/README.md b/README.md
index a8230d3..7f1e268 100644
--- a/README.md
+++ b/README.md
@@ -35,11 +35,11 @@ Please use BertTokenizerFast as tokenizer instead of AutoTokenizer.
 ```
 from transformers import (
   BertTokenizerFast,
-  AutoModelForTokenClassification,
+  AutoModel,
 )
 
 tokenizer = BertTokenizerFast.from_pretrained('bert-base-chinese')
-model = AutoModelForTokenClassification.from_pretrained('ckiplab/bert-base-chinese-ner')
+model = AutoModel.from_pretrained('ckiplab/bert-base-chinese-ner')
 ```
 
 For full usage and more information, please refer to https://github.com/ckiplab/ckip-transformers.