Update app.py

app.py

@@ -20,9 +20,12 @@ model = (
     .to(device)
     .eval()
 )
-def chat_fn(prompt):
+
+
+def respond(message, chat_history):
+    
     #prompt = "Write a python function to sort the following array in ascending order, don't use any built in sorting methods: [9,2,8,1,5]"
-    prompt_input = fmt_prompt(prompt)
+    prompt_input = fmt_prompt(message)
     inputs = tokenizer(prompt_input, return_tensors="pt").to(model.device)
     input_ids_cutoff = inputs.input_ids.size(dim=1)
 
@@ -42,15 +45,17 @@ def chat_fn(prompt):
         generated_ids[0][input_ids_cutoff:],
         skip_special_tokens=True,
     )
-
-    print(completion)
-    return completion
+    chat_history.append((message, completion))
+    time.sleep(2)
+    return "", chat_history
 
 with gr.Blocks() as app:
-    inp = gr.Textbox(label = "Input Prompt")
-    outp = gr.Textbox(label = "Response", lines =6)
-    btn = gr.Button()
 
-    btn.click(chat_fn,inp,outp)
+    chatbot = gr.Chatbot()
+    msg = gr.Textbox("Input")
+    clear = gr.ClearButton([msg, chatbot])
+
+     msg.submit(respond, [msg, chatbot], [msg, chatbot])
+
 
 app.launch()