user message and history

app.py

@@ -68,8 +68,8 @@ def generate_and_tokenize_prompt(data_point):
 #     result["labels"] = result["input_ids"].copy()
     return result
 
-def bot(user_message,temperature, max_new_tokens, top_p,top_k):
-    #user_message = history[-1][0]
+def bot(history,temperature, max_new_tokens, top_p,top_k):
+    user_message = history[-1][0]
     data = {
         'instruction': "Jika Anda seorang dokter, silakan menjawab pertanyaan medis berdasarkan deskripsi pasien.",
         'input': user_message,
@@ -105,11 +105,8 @@ def bot(user_message,temperature, max_new_tokens, top_p,top_k):
     #history[-1] = response.split("Tanggapan:")[1].strip()
     #return history
     response=response.split("Tanggapan:")[1].strip()
-    history[-1][1] = ""
-    for character in response:
-        history[-1][1] += character
-        time.sleep(0.05)
-        yield history
+    history[-1][1] = response
+    return history
 
 
 with gr.Blocks() as demo: