Update app.py

src/app.py

@@ -68,8 +68,30 @@ def load_model(from_pretrained, base_model, cache_dir, device):
                 print("✅ Successfully authenticated with HF token")
             except Exception as e:
                 print(f"⚠️ HF login warning: {e}")
+        
+        # 🔥 新增：从 HF Hub 下载模型
+        # 检查是否是 HF Hub 路径（格式：username/repo-name）
+        is_hf_hub = '/' in from_pretrained and not from_pretrained.startswith('.')
+        if is_hf_hub:
+            from huggingface_hub import snapshot_download
+            print(f"📥 Downloading model from HuggingFace Hub: {from_pretrained}")
+            try:
+                # 下载整个仓库到本地
+                local_model_path = snapshot_download(
+                    repo_id=from_pretrained,
+                    cache_dir=cache_dir,
+                    token=hf_token,
+                    repo_type="model"
+                )
+                print(f"✅ Model downloaded to: {local_model_path}")
+                # 使用下载后的本地路径
+                from_pretrained = local_model_path
+            except Exception as e:
+                print(f"❌ Failed to download model: {e}")
+                raise
     else:
         cache_dir = cache_dir
+    
     with st.spinner("🔄 Loading model... This may take a moment on first launch."):
         model = ComputeScore.from_pretrained(
             from_pretrained,