Update README.md
Browse files
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-
|
152 |
-
model = AutoModelForSequenceClassification.from_pretrained("DAMO-NLP-SG/zero-shot-classify-SSTuning-
|
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"]
|