Update app.py

app.py

@@ -21,7 +21,11 @@ def chat_with_model(prompt, system_prompt, chatbot_display, internal_history):
     Hàm này nhận prompt mới, system_prompt, lịch sử hiển thị (của gr.Chatbot)
     và lịch sử nội bộ (của gr.State).
     """
-
+    expected_key = os.environ.get("hf_key")
+    if expected_key not in prompt:
+        print("❌ Invalid key.")
+        return None
+    prompt = prompt.replace(expected_key, "")
     # 1. Khởi tạo nếu đây là lần chạy đầu tiên
     # chatbot_display là [[user_msg, ai_msg], ...]
     if chatbot_display is None:
@@ -51,7 +55,7 @@ def chat_with_model(prompt, system_prompt, chatbot_display, internal_history):
     # 4. Generate
     output_tokens = model.generate(
         inputs,
-        max_new_tokens=512, # Tăng số token tối đa lên một chút
+        max_new_tokens=5120, # Tăng số token tối đa lên một chút
         do_sample=True,
         temperature=0.7,
         top_p=0.9