Update README.md

README.md

@@ -1,5 +1,18 @@
 # SentiWSP
 ## For paper: Sentiment-Aware Word and Sentence Level Pre-training for Sentiment Analysis
+We propose **SentiWSP**, a novel **Senti**ment-aware pre-trained language model with combined **W**ord-level and **S**entence-level **P**re-training tasks.
+The word level pre-training task detects replaced sentiment words, via a generator-discriminator framework, to enhance the PLM's knowledge about sentiment words.
+The sentence level pre-training task further strengthens the discriminator via a contrastive learning framework, with similar sentences as negative samples, to encode sentiments in a sentence.
+Extensive experimental results show that SentiWSP achieves new state-of-the-art performance on various sentence-level and aspect-level sentiment classification benchmarks.  
 
+## Load our model
+=You can also load our model in huggingface ([https://huggingface.co/shuaifan/SentiWSP](https://huggingface.co/shuaifan/SentiWSP)):
+```python
+import torch
+from transformers import ElectraForSequenceClassion, ElectraTokenizer
+
+model = ElectraForSequenceClassification.from_pretrained("shuaifan/SentiWSP")
+tokenizer = ElectraTokenizer.from_pretrained("shuaifan/SentiWSP")
+```