Update README.md

README.md

@@ -19,12 +19,12 @@ We collect 20  paraphrace datasets in the Chinese domain for finetune, with a to
 
 ## 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-Similarity')
-model=BertForSequenceClassification.from_pretrained('IDEA-CCNL/Erlangshen-MegatronBert-1.3B-Similarity')
+model=AutoModelForSequenceClassification.from_pretrained('IDEA-CCNL/Erlangshen-MegatronBert-1.3B-Similarity')
 
 texta='今天的饭不好吃'
 textb='今天心情不好'