Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 import torch
 from transformers import pipeline, AutoTokenizer
-import os # Thêm thư viện os
+import os
 
 # Lấy token từ secrets của Space
 hf_token = os.environ.get("HF_TOKEN")
@@ -13,13 +13,14 @@ model_id = "phamhoangf/struct-aware-baseline-qwen3-4b"
 # 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
+# ---- SỬA LỖI Ở ĐÂY ----
+# Truyền token trực tiếp cho pipeline, không dùng model_kwargs
 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
+    token=hf_token  # Sửa từ 'model_kwargs' thành cách này
 )
 
 def predict(message, history):