Update README.md
Browse files
README.md
CHANGED
|
@@ -117,7 +117,7 @@ model = AutoModelForSequenceClassification.from_pretrained(
|
|
| 117 |
)
|
| 118 |
model.eval()
|
| 119 |
|
| 120 |
-
pairs = [["中国的首都在哪儿"
|
| 121 |
with torch.no_grad():
|
| 122 |
inputs = tokenizer(pairs, padding=True, truncation=True, return_tensors='pt', max_length=512)
|
| 123 |
scores = model(**inputs, return_dict=True).logits.view(-1, ).float()
|
|
|
|
| 117 |
)
|
| 118 |
model.eval()
|
| 119 |
|
| 120 |
+
pairs = [["中国的首都在哪儿","北京"], ["what is the capital of China?", "北京"], ["how to implement quick sort in python?","Introduction of quick sort"]]
|
| 121 |
with torch.no_grad():
|
| 122 |
inputs = tokenizer(pairs, padding=True, truncation=True, return_tensors='pt', max_length=512)
|
| 123 |
scores = model(**inputs, return_dict=True).logits.view(-1, ).float()
|