assets/download_models.py

import requests
import os
import argparse
from huggingface_hub import snapshot_download
from model_tools import download_files_from_txt, download_hf_repo

REPO_OWNER = "lainlives"
REPO_NAME = "audio-separator-models"
repo_type = "model"


if __name__ == "__main__":
    token = os.getenv("HF_TOKEN")
    repo_id = f"{REPO_OWNER}/{REPO_NAME}"
    parser = argparse.ArgumentParser(description="Model Download Tool")
    local_dir = os.getcwd()
    parser.add_argument(
        "--fromtxt",
        action="store_true",
        help="Download or process based on a text file instead of cloning Hugging Face repo",
    )
    args = parser.parse_args()
    if args.fromtxt:
        download_files_from_txt("assets/links.txt", local_dir)
    else:
        download_hf_repo(repo_id, local_dir, token)