Update README.md

README.md

@@ -8,11 +8,11 @@ Extensive experimental results show that SentiWSP achieves new state-of-the-art
 ## Fine-tunning
 You can also load our model in huggingface ([https://huggingface.co/shuaifan/SentiWSP](https://huggingface.co/shuaifan/SentiWSP)):
 ```python
+from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import torch
-from transformers import ElectraForSequenceClassion, ElectraTokenizer
+tokenizer = AutoTokenizer.from_pretrained("shuaifan/SentiWSP")
 
-model = ElectraForSequenceClassification.from_pretrained("shuaifan/SentiWSP")
-tokenizer = ElectraTokenizer.from_pretrained("shuaifan/SentiWSP")
+model = AutoModelForSequenceClassification.from_pretrained("shuaifan/SentiWSP")
 ```