Added a functionality in UI

app.py

@@ -153,5 +153,12 @@ with gr.Blocks() as demo:
         inputs=[msg, chatbot, temperature, max_tokens],
         outputs=[chatbot, chatbot, context_box]
     )
+    
+    msg.submit(
+    chat_wrapper,
+    inputs=[msg, chatbot, temperature, max_tokens],
+    outputs=[chatbot, chatbot, context_box]
+)
+
 
 demo.launch()