From 378508b83b468422197f285a65606abf7d0fd5e9 Mon Sep 17 00:00:00 2001 From: XiaoqianShen Date: Mon, 17 Apr 2023 10:35:02 +0300 Subject: [PATCH] Update demo.py --- demo.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/demo.py b/demo.py index 8ceea3f..4a23e80 100644 --- a/demo.py +++ b/demo.py @@ -66,13 +66,15 @@ print('Initialization Finished') # ======================================== def gradio_reset(chat_state, img_list): - chat_state.messages = [] - img_list = [] + if chat_state is not None: + chat_state.messages = [] + if img_list is not None: + img_list = [] return None, gr.update(value=None, interactive=True), gr.update(placeholder='Please upload your image first', interactive=False),gr.update(value="Upload & Start Chat", interactive=True), chat_state, img_list def upload_img(gr_img, text_input, chat_state): if gr_img is None: - return None, None, gr.update(interactive=True) + return None, None, gr.update(interactive=True), chat_state, None chat_state = CONV_VISION.copy() img_list = [] llm_message = chat.upload_img(gr_img, chat_state, img_list) @@ -93,9 +95,7 @@ def gradio_answer(chatbot, chat_state, img_list, num_beams, temperature): title = """

Demo of MiniGPT-4

""" description = """

This is the demo of MiniGPT-4. Upload your images and start chatting!

""" -article = """Paper: Here -Code: Here -Project Page: Here +article = """

""" #TODO show examples below