Update README.md

README.md

@@ -53,13 +53,13 @@ model_trained.to(device)
 def postprocess(text):
   return text.replace(".", "").replace('</>','')
 
-def answer_fn(text, sample=False, top_p=0.6):
+def answer_fn(text, top_p=0.6):
   encoding = tokenizer(text=[text], truncation=True, padding=True, max_length=256, return_tensors="pt").to(device) 
   out = model.generate(**encoding, return_dict_in_generate=True, output_scores=False, max_length=512,temperature=0.5,do_sample=True,repetition_penalty=6.0 ,top_p=top_p)
   result = tokenizer.batch_decode(out["sequences"], skip_special_tokens=True)
   return postprocess(result[0]) 
 text="宫颈癌的早期会有哪些危险信号"
-result=answer_fn(text, sample=True, top_p=0.6)
+result=answer_fn(text, top_p=0.6)
 print('prompt:',text)
 print("result:",result)
 ```