Update README snippets

README.md

@@ -37,7 +37,7 @@ scores = model.predict([
     ("How many people live in Berlin?", "Berlin is well known for its museums."),
 ])
 print(scores)
-# [ 8.607138 -4.320078]
+# [ 8.122868 -4.292924]
 ```
 
 ## Usage with Transformers
@@ -51,8 +51,8 @@ tokenizer = AutoTokenizer.from_pretrained("cross-encoder/monoelectra-base")
 
 features = tokenizer(
     [
-        ["How many people live in Berlin?", "How many people live in Berlin?"],
-        ["Berlin has a population of 3,520,031 registered inhabitants in an area of 891.82 square kilometers.", "New York City is famous for the Metropolitan Museum of Art."],
+        ("How many people live in Berlin?", "Berlin had a population of 3,520,031 registered inhabitants in an area of 891.82 square kilometers."),
+        ("How many people live in Berlin?", "Berlin is well known for its museums."),
     ],
     padding=True,
     truncation=True,
@@ -61,6 +61,7 @@ features = tokenizer(
 
 model.eval()
 with torch.no_grad():
-    scores = model(**features).logits
-    print(scores)
+    scores = model(**features).logits.view(-1)
+print(scores)
+# tensor([ 8.1229, -4.2929])
 ```