scripts/create_hf_package.py

#!/usr/bin/env python3
"""Create a local Hugging Face Hub package from a trained model."""

from __future__ import annotations

import argparse
import sys
from pathlib import Path

PROJECT_ROOT = Path(__file__).resolve().parent.parent
sys.path.insert(0, str(PROJECT_ROOT))

from src.huggingface import package_for_hub

def main() -> None:
    parser = argparse.ArgumentParser(description="Create HF Hub package locally")
    parser.add_argument(
        "--model",
        type=Path,
        default=PROJECT_ROOT / "models" / "multi_tower_dev.joblib",
    )
    parser.add_argument(
        "--output",
        type=Path,
        default=PROJECT_ROOT / "models" / "hf_package",
    )
    args = parser.parse_args()

    out = package_for_hub(args.model, args.output)
    print(f"Package ready at {out}")
    print("Files:", [p.name for p in out.iterdir()])


if __name__ == "__main__":
    main()