mkw18 commited on
Commit
495ab3c
·
1 Parent(s): f997697
Files changed (1) hide show
  1. app.py +8 -2
app.py CHANGED
@@ -80,9 +80,11 @@ def predict(input, chatbot, messages, idx, answer, story_key, answer_key, known,
80
  print(f"Start judge: {time.time() - start}")
81
  for key in story_key:
82
  key = key.strip()
 
 
83
  if ' ' in key:
84
  key = key.split(' ')[1]
85
- bleu = sentence_bleu(key, input.replace('?', '。'), weights=(1, 0, 0, 0))
86
  if bleu > 0.85:
87
  response = '这是汤面中已有的信息,请提一个新问题。'
88
  llm = False
@@ -91,9 +93,11 @@ def predict(input, chatbot, messages, idx, answer, story_key, answer_key, known,
91
  if llm:
92
  for key in history:
93
  key = key.strip()
 
 
94
  if ' ' in key:
95
  key = key.split(' ')[1]
96
- bleu = sentence_bleu(key, input.replace('?', '。'), weights=(1, 0, 0, 0))
97
  if bleu > 0.85:
98
  response = '这是已经提问过的内容,请提一个新问题。'
99
  llm = False
@@ -170,6 +174,8 @@ def predict(input, chatbot, messages, idx, answer, story_key, answer_key, known,
170
  merge = summary
171
  for key in answer_key:
172
  key = key.strip()
 
 
173
  if ' ' in key:
174
  key1 = key.split(' ')[1]
175
  else:
 
80
  print(f"Start judge: {time.time() - start}")
81
  for key in story_key:
82
  key = key.strip()
83
+ if key == '':
84
+ continue
85
  if ' ' in key:
86
  key = key.split(' ')[1]
87
+ bleu = sentence_bleu([key], input.replace('?', '。'), weights=(1, 0, 0, 0))
88
  if bleu > 0.85:
89
  response = '这是汤面中已有的信息,请提一个新问题。'
90
  llm = False
 
93
  if llm:
94
  for key in history:
95
  key = key.strip()
96
+ if key == '':
97
+ continue
98
  if ' ' in key:
99
  key = key.split(' ')[1]
100
+ bleu = sentence_bleu([key], input.replace('?', '。'), weights=(1, 0, 0, 0))
101
  if bleu > 0.85:
102
  response = '这是已经提问过的内容,请提一个新问题。'
103
  llm = False
 
174
  merge = summary
175
  for key in answer_key:
176
  key = key.strip()
177
+ if key == '':
178
+ continue
179
  if ' ' in key:
180
  key1 = key.split(' ')[1]
181
  else: