Update README.md

README.md

@@ -19,7 +19,7 @@ references = ["hello world", "hello world"]
 candidates = ["hi universe", "bye world"]
 
 with torch.no_grad():
-  scores = model(**tokenizer(references, candidates, return_tensors='pt'))[0]
+  scores = model(**tokenizer(references, candidates, return_tensors='pt'))[0].squeeze()
 
-print(scores) # tensor([[0.9877], [0.0475]])
+print(scores) # tensor([1.0327, 0.2055])
 ```