Update app.py

app.py

@@ -41,9 +41,9 @@ def generate(
     for response in stream:
         output += response.token.text
         yield output
-    return output
+    return output,output
 
-output = gr.Chatbot(show_label=False, show_share_button=False, show_copy_button=True, likeable=True, layout="panel")
+output = [gr.Chatbot(show_label=False, show_share_button=False, show_copy_button=True, likeable=True, layout="panel"),chat_state]
 additional_inputs=[
     gr.Textbox(
         label="Prompt",