Update app.py

app.py

@@ -45,6 +45,7 @@ HEADERS = {
 
 
 MAX_SEC = 30
+MAX_INPUT_LENGTH = 5000
 
 tokenizer = AutoTokenizer.from_pretrained("MediaTek-Research/Breeze-7B-Instruct-v0_1")
 
@@ -161,6 +162,9 @@ with gr.Blocks() as demo:
         message = tokenizer.apply_chat_template(chat_data, tokenize=False)
         message = message[3:]  # remove SOT token
 
+        if len(message) > MAX_INPUT_LENGTH:
+            raise Exception()
+
         response = ''
         if refusal_condition(history[-1][0]):
             history = [['[安全拒答啟動]', '[安全拒答啟動] 請清除再開啟對話']]