lukecq commited on
Commit
d66431a
1 Parent(s): 1265437

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -148,8 +148,8 @@ You can try the model with the Colab [Notebook](https://colab.research.google.co
148
  from transformers import AutoTokenizer, AutoModelForSequenceClassification
149
  import torch, string, random
150
 
151
- tokenizer = AutoTokenizer.from_pretrained("DAMO-NLP-SG/zero-shot-classify-SSTuning-base")
152
- model = AutoModelForSequenceClassification.from_pretrained("DAMO-NLP-SG/zero-shot-classify-SSTuning-base")
153
 
154
  text = "I love this place! The food is always so fresh and delicious."
155
  list_label = ["negative", "positive"]
 
148
  from transformers import AutoTokenizer, AutoModelForSequenceClassification
149
  import torch, string, random
150
 
151
+ tokenizer = AutoTokenizer.from_pretrained("DAMO-NLP-SG/zero-shot-classify-SSTuning-XLM-R")
152
+ model = AutoModelForSequenceClassification.from_pretrained("DAMO-NLP-SG/zero-shot-classify-SSTuning-XLM-R")
153
 
154
  text = "I love this place! The food is always so fresh and delicious."
155
  list_label = ["negative", "positive"]