Update app.py

app.py

@@ -1,16 +1,25 @@
 import gradio as gr
 import torch
 from transformers import pipeline, AutoTokenizer
+import os # Thêm thư viện os
+
+# Lấy token từ secrets của Space
+hf_token = os.environ.get("HF_TOKEN")
 
 # Tải mô hình và tokenizer
 # device_map="auto" sẽ tự động sử dụng GPU nếu có
 model_id = "phamhoangf/struct-aware-baseline-qwen3-4b"
-tokenizer = AutoTokenizer.from_pretrained(model_id)
+
+# Sử dụng token để xác thực khi tải tokenizer
+tokenizer = AutoTokenizer.from_pretrained(model_id, token=hf_token)
+
+# Sử dụng token để xác thực khi tải model thông qua pipeline
 pipe = pipeline(
     "text-generation",
     model=model_id,
     torch_dtype=torch.bfloat16,
     device_map="auto",
+    model_kwargs={"token": hf_token} # Rất quan trọng: truyền token cho model
 )
 
 def predict(message, history):