Update README.md

README.md

@@ -19,12 +19,12 @@ We collect 4 NLI（Natural Language Inference） datasets in the Chinese domain
 
 ## Usage
 ```python
-from transformers import BertForSequenceClassification
+from transformers import AutoModelForSequenceClassification
 from transformers import BertTokenizer
 import torch
 
 tokenizer=BertTokenizer.from_pretrained('IDEA-CCNL/Erlangshen-MegatronBert-1.3B-NLI')
-model=BertForSequenceClassification.from_pretrained('IDEA-CCNL/Erlangshen-MegatronBert-1.3B-NLI')
+model=AutoModelForSequenceClassification.from_pretrained('IDEA-CCNL/Erlangshen-MegatronBert-1.3B-NLI')
 
 texta='今天的饭不好吃'
 textb='今天心情不好'