Update README.md

README.md

@@ -93,8 +93,10 @@ tokenizer = AutoTokenizer.from_pretrained('namdp/bge-reranker-vietnamese')
 model = AutoModelForSequenceClassification.from_pretrained('namdp/bge-reranker-vietnamese')
 model.eval()
 
-pairs = [['what is panda?', 'hi'], ['what is panda?',
-                                    'The giant panda (Ailuropoda melanoleuca), sometimes called a panda bear or simply panda, is a bear species endemic to China.']]
+pairs = [
+    ['tỉnh nào có diện tích lớn nhất việt nam', 'nghệ an có diện tích lớn nhất việt nam'],
+    ['tỉnh nào có diện tích lớn nhất việt nam', 'bắc ninh có diện tích nhỏ nhất việt nam']
+],
 with torch.no_grad():
     inputs = tokenizer(pairs, padding=True, truncation=True, return_tensors='pt', max_length=512)
     scores = model(**inputs, return_dict=True).logits.view(-1, ).float()