update probs readme

README.md

@@ -138,11 +138,14 @@ inputs = tokenizer(
             "query 1",
             "passage_11",
         ],
+        [
+            "query 1",
+            "passage_12",
+        ],
         [
             "query_2",
             "passage_21",
-        ]
-        
+        ],
     ],
     return_tensors="pt",
     padding=True,
@@ -162,14 +165,18 @@ inputs = [
     "query 1",
     "passage_11",
   ],
+  [
+    "query 1",
+    "passage_12",
+  ],
   [
     "query_2",
     "passage_21",
-  ]
+  ],
 ]
 scores = model.predict(inputs)
 print(scores)
-# [0.4359, 0.3840]
+# [0.43585014, 0.32305932, 0.38395187]
 ```
 
 To reproduce the results with iterative inference, please run: