Update README.md

README.md

@@ -24,14 +24,14 @@ if paragraph is not None:
 prompt += "[Retrieval]<paragraph>{0}</paragraph>".format(paragraph)
 return prompt
 
-query_1 = "你好"
-query_2 = "世界最高的山峰是什么？"
+query_1 = "你好呀"
+query_2 = "故宫三大殿是哪些？"
 queries = [query_1, query_2]
 
 preds = model.generate([format_prompt(query) for query in queries], sampling_params)
 for pred in preds:
 print("Model prediction: {0}".format(pred.outputs[0].text))
-# Model prediction:
-
+# Model prediction: [No Retrieval] 你好！有什么我可以帮你解答的问题吗？ [Utility:5] </s>
+# Model prediction: [Relevant] 太和殿、中和殿、保和殿 [Utility:5] </s>
 ```