Update app.py

app.py

@@ -116,7 +116,6 @@ with gr.Blocks() as demo:
         message = tokenizer.apply_chat_template(chat_data, tokenize=False)
         message = message[3:]  # remove SOT token
 
-        print('request:', repr(message))
         
         data = {
             "model": MODEL_NAME,
@@ -147,9 +146,11 @@ with gr.Blocks() as demo:
                         else:
                             history[-1][1] += delta
                         yield history
-            if history[-1][1].endswith('</s>'):
-                history[-1][1] = history[-1][1][:-4]
-                yield history
+        if history[-1][1].endswith('</s>'):
+            history[-1][1] = history[-1][1][:-4]
+            yield history
+
+        print('== Record == Query: {query}\n{response}'.format(query=repr(message), response=repr(history[-1][1])))
 
     msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
         fn=bot,