namdp-ptit
/

ViRanker

Text Classification

text-embeddings-inference

Model card Files Files and versions

namdp-ptit commited on Aug 19, 2024

Commit

bc8d82d

·

verified ·

1 Parent(s): 524a63b

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -94,8 +94,8 @@ model = AutoModelForSequenceClassification.from_pretrained('namdp-ptit/ViRanker'
 model.eval()
 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)

 model.eval()
 pairs = [
+    ['ai là vị vua cuối cùng của việt nam', 'vua bảo đại là vị vua cuối cùng của nước ta'],
+    ['ai là vị vua cuối cùng của việt nam', 'lý nam đế là vị vua đầu tiên của nước ta']
 ],
 with torch.no_grad():
     inputs = tokenizer(pairs, padding=True, truncation=True, return_tensors='pt', max_length=512)