Update README.md

README.md

@@ -19,12 +19,12 @@ We collect 8 sentiment datasets in the Chinese domain for finetune, with a tota
 
 ## 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-Sentiment')
-model=BertForSequenceClassification.from_pretrained('IDEA-CCNL/Erlangshen-MegatronBert-1.3B-Sentiment')
+model=AutoModelForSequenceClassification.from_pretrained('IDEA-CCNL/Erlangshen-MegatronBert-1.3B-Sentiment')
 
 text='今天心情不好'