Update app.py

app.py

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