Fix bug in loading history

This commit is contained in:
oobabooga 2023-01-22 02:32:54 -03:00
parent 62b533f344
commit b3e1a874bc

View File

@ -348,7 +348,8 @@ if args.chat or args.cai_chat:
def redraw_html(name1, name2): def redraw_html(name1, name2):
global history global history
return generate_chat_html(history, name1, name2, character) _history = remove_example_dialogue_from_history(history)
return generate_chat_html(_history, name1, name2, character)
def save_history(): def save_history():
if not Path('logs').exists(): if not Path('logs').exists():
@ -478,7 +479,7 @@ if args.chat or args.cai_chat:
if args.cai_chat: if args.cai_chat:
upload.upload(redraw_html, [name1, name2], [display1]) upload.upload(redraw_html, [name1, name2], [display1])
else: else:
upload.upload(lambda : history, [], [display1]) upload.upload(lambda : remove_example_dialogue_from_history(history), [], [display1])
elif args.notebook: elif args.notebook:
with gr.Blocks(css=css, analytics_enabled=False) as interface: with gr.Blocks(css=css, analytics_enabled=False) as interface: