assets/upload_models.py

#! /usr/bin/env python3
import argparse
import os

from huggingface_hub import HfApi

import model_tools as mt

REPO_OWNER = "lainlives"
REPO_NAME = "audio-separator-models"
repo_type = "model"
HF_TOKEN = os.getenv("HF_TOKEN")


if __name__ == "__main__":
    repo_id = f"{REPO_OWNER}/{REPO_NAME}"

    parser = argparse.ArgumentParser(description="Model Upload Tool")
    parser.add_argument(
        "--large",
        action="store_true",
        help="Upload using upload-large-folder",
    )
    parser.add_argument(
        "--token",
        type=str,
        default=HF_TOKEN,
        dest="HF_TOKEN",
        help=f"Your access token.",
    )
    args = parser.parse_args()
    if args.large:
        mt.push_large_folder_to_hf(repo_id, repo_type)
    else:
        mt.push_to_hf(repo_id, repo_type)