AllenYkl commited on
Commit
53d92c9
·
1 Parent(s): 7a6bd30

Update bin_public/utils/utils.py

Browse files
Files changed (1) hide show
  1. bin_public/utils/utils.py +4 -8
bin_public/utils/utils.py CHANGED
@@ -320,14 +320,10 @@ def get_template_content(templates, selection, original_system_prompt):
320
  return original_system_prompt
321
 
322
  def get_pre_defined_q(selection):
323
- q1 = '医生您好,从昨天晚上开始 肚子一直疼, 该吃什么药呢'
324
- q2 = '腹泻应该用什么药'
325
- q3 = '我四肢冰凉,应该用什么药'
326
- q4 = '藿香正气口服液有什么功效'
327
- q5 = '藿香正气口服液都适合什么时候服用'
328
- q6 = '夏天到了,我吃火锅的时候,可以服用藿香正气口服液么'
329
- q7 = '我吃火锅上火,喝藿香正气口服液有用么'
330
- qs = [q1, q2, q3, q4, q5, q6, q7]
331
  return qs[qs.index(selection)]
332
 
333
  def reset_state():
 
320
  return original_system_prompt
321
 
322
  def get_pre_defined_q(selection):
323
+ with open(r'\questions.txt', 'r') as f:
324
+ qs = f.readlines()
325
+ for i in qs:
326
+ qs[qs.index(i)] = i.replace('\n', '')
 
 
 
 
327
  return qs[qs.index(selection)]
328
 
329
  def reset_state():