将处理后的元素添加到新的列表中,避免了每次循环都要修改原列表的操作,提高了效率

pull/1325/head
SuZui-cn 2023-07-10 21:25:47 +08:00
parent 16fd212c05
commit 5024bee528
1 changed files with 6 additions and 6 deletions

View File

@ -12,12 +12,12 @@ model = model.eval()
def postprocess(self, y):
if y is None:
return []
for i, (message, response) in enumerate(y):
y[i] = (
None if message is None else mdtex2html.convert((message)),
None if response is None else mdtex2html.convert(response),
)
return y
processed_y = []
for message, response in y:
processed_message = None if message is None else mdtex2html.convert(message)
processed_response = None if response is None else mdtex2html.convert(response)
processed_y.append((processed_message, processed_response))
return processed_y
gr.Chatbot.postprocess = postprocess