Update app.py

app.py

@@ -3,6 +3,14 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
 import hashlib
 import logging
+from huggingface_hub import login
+import os
+
+# Lấy token từ secrets trong Hugging Face Spaces
+HF_TOKEN = os.environ.get("HF_TOKEN")  # "HF_TOKEN" là tên biến bạn đặt trong Secrets
+if not HF_TOKEN:
+    raise ValueError("HF_TOKEN not found in environment variables. Please set it in Hugging Face Spaces Secrets.")
+login(token=HF_TOKEN)
 
 # Set up logging
 logging.basicConfig(filename="chatbot.log", level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
@@ -91,7 +99,10 @@ III. Company General Info
 system_prompt = f"{system_persona}\nHere is the knowledge base for your reference:\n{knowledge_base}\nAnswer the user's question based on the information above, following the constraints and communication style outlined in the persona."
 
 # Load Gemma-2B model và tokenizer
-tokenizer = AutoTokenizer.from_pretrained("google/gemma-2b", padding_side="left")
+tokenizer = AutoTokenizer.from_pretrained(
+    "google/gemma-2b",
+    padding_side="left"
+)
 tokenizer.pad_token = tokenizer.eos_token
 
 model = AutoModelForCausalLM.from_pretrained(