Update app.py

app.py

@@ -158,8 +158,11 @@ with gr.Blocks() as demo:
                 chat_data.append({"role": "user", "content": user_msg})
             if assistant_msg is not None:
                 chat_data.append({"role": "assistant", "content": assistant_msg})
-
-        message = tokenizer.apply_chat_template(chat_data, tokenize=False)
+        try:
+            message = tokenizer.apply_chat_template(chat_data, tokenize=False)
+        except Exception as e:
+            print(chat_data)
+            raise Exception('prompt template error')
         message = message[3:]  # remove SOT token
 
         if len(message) > MAX_INPUT_LENGTH: