update app.py

app.py

@@ -20,6 +20,14 @@ REPO_ID = os.environ.get("ASSETS_REPO_ID", "pixai-labs/pixai-tagger-v0.9")
 REVISION = os.environ.get("ASSETS_REVISION")  # optional pin, e.g. "main" or a commit
 MODEL_DIR = os.environ.get("MODEL_DIR", "./assets")  # where the handler will look
 
+# Optional: Hugging Face token for private repos
+HF_TOKEN = (
+    os.environ.get("HUGGINGFACE_HUB_TOKEN")
+    or os.environ.get("HF_TOKEN")
+    or os.environ.get("HUGGINGFACE_TOKEN")
+    or os.environ.get("HUGGINGFACEHUB_API_TOKEN")
+)
+
 REQUIRED_FILES = [
     "model_v0.9.pth",
     "tags_v0.9_13k.json",
@@ -44,6 +52,7 @@ def ensure_assets(repo_id: str, revision: Optional[str], target_dir: str):
         repo_id=repo_id,
         revision=revision,
         allow_patterns=REQUIRED_FILES,  # only pull what we need
+        token=HF_TOKEN,  # authenticate if repo is private
     )
 
     # Copy files into target_dir with the required names