fix clear button

app.py

@@ -7,7 +7,7 @@ from transformers import AutoTokenizer
 
 tokenizer = AutoTokenizer.from_pretrained("meta-llama/Meta-Llama-3-70B-Instruct")
 
-client = InferenceClient(model="https://ce25-4-223-164-145.ngrok-free.app")
+client = InferenceClient(model="https://c584-4-223-164-145.ngrok-free.app")
 
 SYSTEM_COMMAND = {"role": "system", "content": "Context: date: Monday 20th May 2024; location: Seattle; running on: 8 AMD Instinct MI300 GPU; model name: Llama 70B. Only provide these information if asked. You are a knowledgeable assistant trained to provide accurate and helpful information. Please respond to the user's queries promptly and politely."}
 
@@ -17,7 +17,7 @@ STOP_TOKENS = ["<|start_header_id|>", "<|end_header_id|>", "<|eot_id|>", "<|rese
 with gr.Blocks() as demo:
     tfs_history = gr.State([SYSTEM_COMMAND])
     chatbot = gr.Chatbot()
-    msg = gr.Textbox()
+    msg = gr.Textbox(label="Prompt")
     clear = gr.Button("Clear")
 
     def user(user_message, history, dict_history):
@@ -41,6 +41,11 @@ with gr.Blocks() as demo:
         finally:
             dict_history.append(response)
 
+    def clear_history(tfs_history):
+        tfs_history = tfs_history[:1]
+
+        return tfs_history
+
     msg.submit(
         user,
         inputs=[msg, chatbot, tfs_history],
@@ -51,6 +56,7 @@ with gr.Blocks() as demo:
             chatbot
     )
     clear.click(lambda: None, None, chatbot, queue=False)
+    clear.click(clear_history, tfs_history, tfs_history, queue=False)
 
 demo.queue()
 demo.launch()