update code

requirements.txt

@@ -7,4 +7,5 @@ numpy==2.1.3
 transformers==4.55.2
 peft==0.17.1
 tqdm
-scikit-learn
+scikit-learn
+huggingface_hub

src/app.py

@@ -28,6 +28,18 @@ def load_model(from_pretrained, base_model, cache_dir, device):
         cache_dir = str(Path.home() / '.cache' / 'huggingface')
         device = 'cpu'
         print("Using **CPU** now!")
+
+        # 获取 HF Token（用于访问 gated 模型）
+        hf_token = os.environ.get('HF_TOKEN', None)
+        if hf_token:
+            kwargs['token'] = hf_token
+            # 也可以用 login 方式
+            try:
+                from huggingface_hub import login
+                login(token=hf_token)
+                print("✅ Successfully authenticated with HF token")
+            except Exception as e:
+                print(f"⚠️ HF login warning: {e}")
     else:
         cache_dir = cache_dir
     with st.spinner("🔄 Loading model... This may take a moment on first launch."):