Update app.py

app.py

@@ -39,13 +39,13 @@ def predict(
     top_k: int,
     repetition_penalty: float,
     watermark: bool,
-    # chatbot,
-    # history,
+    chatbot,
+    history,
 ):
     client = get_client(model)
     preprompt, user_name, assistant_name, sep = get_usernames(model)
 
-    # history.append(inputs)
+    history.append(inputs)
 
     past = []
     for data in chatbot:
@@ -96,16 +96,16 @@ def predict(
         if partial_words.endswith(assistant_name.rstrip()):
             partial_words = partial_words.rstrip(assistant_name.rstrip())
 
-        # if i == 0:
-        #     history.append(" " + partial_words)
-        # elif response.token.text not in user_name:
-        #     history[-1] = partial_words
+        if i == 0:
+            history.append(" " + partial_words)
+        elif response.token.text not in user_name:
+            history[-1] = partial_words
 
-        # chat = [
-        #     (history[i].strip(), history[i + 1].strip())
-        #     for i in range(0, len(history) - 1, 2)
+        chat = [
+            (history[i].strip(), history[i + 1].strip())
+            for i in range(0, len(history) - 1, 2)
         ]
-        yield chat#, history
+        yield chat, history
 
 
 def reset_textbox():
@@ -291,8 +291,8 @@ with gr.Blocks(
             top_k,
             repetition_penalty,
             watermark,
-            # chatbot,
-            # state,
+            chatbot,
+            state,
         ],
         [chatbot, state],
         api_name = "chat_text",
@@ -308,8 +308,8 @@ with gr.Blocks(
             top_k,
             repetition_penalty,
             watermark,
-            # chatbot,
-            # state,
+            chatbot,
+            state,
         ],
         [chatbot, state],
         api_name = "chat_button",