update model_download

download_model.py

@@ -1,14 +1,19 @@
+#
+from huggingface_hub import login, snapshot_download
 import os
-from huggingface_hub import snapshot_download 
-hf_token = os.getenv("HF_TOKEN") 
-
 MODEL_REPO = "facebook/mobilellm-pro"  # ← YOUR model
 LOCAL_DIR = "/app/model"
-##
-print(f"📥 Downloading model from {MODEL_REPO}...")
+
+# Log in using the token from environment variable
+hf_token = os.getenv("HF_TOKEN")
+if hf_token:
+    login(token=hf_token)
+else:
+    print("⚠️ Warning: HF_TOKEN not found. Make sure it's set in your environment.")
+
+# Now download the model
 snapshot_download(
     repo_id=MODEL_REPO,
     local_dir=LOCAL_DIR,
-    local_dir_use_symlinks=False  # avoids symlink issues in Docker
+    token=hf_token  # explicitly pass it
 )
-print("✅ Model downloaded successfully!")