update

Dockerfile

@@ -1,5 +1,7 @@
 FROM python:3.10.8
 
+python download_private_model.py
+
 WORKDIR /app
 
 RUN apt-get update && apt-get install -y \

download_private_code.py

@@ -0,0 +1,37 @@
+# src/download_private_model.py
+import os
+import requests
+from pathlib import Path
+
+MODEL_DIR = Path("/src/FineTune")
+MODEL_PATH = MODEL_DIR / "model.py"
+
+def download_model():
+    print("📥 Checking private model file...")
+    if MODEL_PATH.exists():
+        print(f"✅ model.py already exists at {MODEL_PATH}")
+        return
+
+    hf_token = os.getenv("HF_TOKEN")
+    if not hf_token:
+        raise RuntimeError(
+            "❌ HF_TOKEN not found. Please set it in Hugging Face Space → Settings → Secrets."
+        )
+
+    url = "https://huggingface.co/mamba413/ada-core/resolve/main/model.py"
+    headers = {"Authorization": f"Bearer {hf_token}"}
+    MODEL_DIR.mkdir(parents=True, exist_ok=True)
+
+    print(f"📡 Downloading from {url}")
+    response = requests.get(url, headers=headers)
+    if response.status_code != 200:
+        raise RuntimeError(
+            f"❌ Failed to download model.py (HTTP {response.status_code}): {response.text[:200]}"
+        )
+
+    with open(MODEL_PATH, "wb") as f:
+        f.write(response.content)
+    print(f"✅ Successfully downloaded model.py to {MODEL_PATH}")
+
+if __name__ == "__main__":
+    download_model()

src/app.py

@@ -1,7 +1,5 @@
 import os
 from pathlib import Path
-import requests
-import sys
 
 # -----------------
 # Get the directory where app.py is located
@@ -30,39 +28,6 @@ if os.environ.get('SPACE_ID'):
     streamlit_dir = Path('/tmp/.streamlit')
     streamlit_dir.mkdir(exist_ok=True, parents=True)
 
-# ======================================================
-# ✅ 新增：在运行时从 Hugging Face 下载私有 model.py（写入 /tmp）
-# ======================================================
-target_dir = Path("/tmp/FineTune")
-target_path = target_dir / "model.py"
-
-if not target_path.exists():
-    print(f"📥 Attempting to download private model.py to: {target_path}")
-    hf_token = os.getenv("HF_TOKEN")
-    if not hf_token:
-        raise RuntimeError(
-            "❌ HF_TOKEN not found. Please configure it in Hugging Face Space → Settings → Secrets."
-        )
-
-    url = "https://huggingface.co/mamba413/ada-core/resolve/main/model.py"
-    headers = {"Authorization": f"Bearer {hf_token}"}
-    response = requests.get(url, headers=headers)
-
-    if response.status_code != 200:
-        raise RuntimeError(
-            f"❌ Failed to download model.py (HTTP {response.status_code}): {response.text[:200]}"
-        )
-
-    os.makedirs(target_dir, exist_ok=True)
-    with open(target_path, "wb") as f:
-        f.write(response.content)
-    print("✅ Successfully downloaded private model.py")
-else:
-    print("✅ model.py already exists in /tmp, skipping download.")
-
-# ✅ 添加 /tmp 到 Python 导入路径
-if str(target_dir.parent) not in sys.path:
-    sys.path.insert(0, str(target_dir.parent))
 
 import streamlit as st
 from FineTune.model import ComputeScore