diff --git a/demo.py b/demo.py
index 2027b96..16e5f3b 100644
--- a/demo.py
+++ b/demo.py
@@ -165,7 +165,8 @@ with gr.Blocks() as demo:
                         [image, text_input, upload_button, chat_state, emb_list])
 
     text_input.submit(gradio_ask, [text_input, chatbot, chat_state], [text_input, chatbot, chat_state]).then(
-        image, upload_img, gradio_answer, [chatbot, chat_state, emb_list, num_beams, temperature], [image2, chatbot, chat_state, emb_list]
+        upload_img, gradio_answer, [image, chatbot, chat_state, emb_list, num_beams, temperature],
+        [image2, chatbot, chat_state, emb_list]
     )
     clear.click(gradio_reset, [chat_state, emb_list], [chatbot, image, text_input, upload_button, chat_state, emb_list],
                 queue=False)