mirror of https://github.com/THUDM/ChatGLM-6B
将处理后的元素添加到新的列表中,避免了每次循环都要修改原列表的操作,提高了效率
parent
16fd212c05
commit
5024bee528
12
web_demo.py
12
web_demo.py
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue