Update download_model.py

download_model.py

@@ -1,19 +1,17 @@
-#
 from huggingface_hub import login, snapshot_download
 import os
-MODEL_REPO = "facebook/mobilellm-pro"  # ← YOUR model
+
+MODEL_REPO = "facebook/mobilellm-pro"
 LOCAL_DIR = "/app/model"
 
-# 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,
-    token=hf_token  # explicitly pass it
+    token=hf_token
 )