from huggingface_hub import login, snapshot_download
import os

MODEL_REPO = "facebook/mobilellm-pro"
LOCAL_DIR = "/app/model"

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.")

snapshot_download(
    repo_id=MODEL_REPO,
    local_dir=LOCAL_DIR,
    token=hf_token
)