Update README.md

README.md

@@ -48,7 +48,7 @@ def preprocess(text):
 def postprocess(text):
   return text.replace("\\n", "\n").replace("\\t", "\t")
 
-def answer(text, sample=True, top_p=0.9):
+def answer(text, sample=True, top_p=1, temperature=0.7):
   '''sample：是否抽样。生成任务，可以设置为True;
   top_p：0-1之间，生成的内容越多样'''
   text = preprocess(text)
@@ -56,9 +56,10 @@ def answer(text, sample=True, top_p=0.9):
   if not sample:
     out = model.generate(**encoding, return_dict_in_generate=True, output_scores=False, max_new_tokens=512, num_beams=1, length_penalty=0.6)
   else:
-    out = model.generate(**encoding, return_dict_in_generate=True, output_scores=False, max_new_tokens=512, do_sample=True, top_p=top_p, no_repeat_ngram_size=3)
+    out = model.generate(**encoding, return_dict_in_generate=True, output_scores=False, max_new_tokens=512, do_sample=True, top_p=top_p, temperature=temperature, no_repeat_ngram_size=3)
   out_text = tokenizer.batch_decode(out["sequences"], skip_special_tokens=True)
   return postprocess(out_text[0])
+print("end...")
 ```
 
 # 问答、写作与功能型助手