Update README.md

README.md

@@ -68,19 +68,18 @@ blender.loadranker("llm-blender/PairRM") # load PairRM
 - Ranking a list candidate responses
 
 ```python
-inputs = ["hello!", "I love you!"]
-candidates_texts = [["get out!", "hi! nice to meet you!", "bye"], 
+inputs = ["hello, how are you!", "I love you!"]
+candidates_texts = [["get out!", "hi! I am fine, thanks!", "bye!"], 
                     ["I love you too!", "I hate you!", "Thanks! You're a good guy!"]]
-ranks = blender.rank(inputs, candidates_texts, return_scores=False, batch_size=2)
-# ranks is a list of ranks where
+ranks = blender.rank(inputs, candidates_texts, return_scores=False, batch_size=1)
+# ranks is a list of ranks
 # ranks[i][j] represents the ranks of candidate-j for input-i
 """
 ranks -->
-array([[3, 1, 2], # it means "hi! nice to meet you!" ranks the 1st,
-                      "bye" ranks the 2nd, and "get out!" ranks the 3rd. 
-       [1, 3, 2]], # it means "I love you too"! ranks the the 1st,
-                    and "I hate you!" ranks the 3rd.
-       dtype=int32)
+array([[3, 1, 2], # it means "hi! I am fine, thanks!" ranks the 1st, "bye" ranks the 2nd, and "get out!" ranks the 3rd. 
+       [1, 3, 2]], # it means "I love you too"! ranks the the 1st, and "I hate you!" ranks the 3rd.
+       dtype=int32) 
+
 """
 ```