import os
import shutil
import glob

def copy_models():
    src_dir = "models"
    dest_dir = "hf_deploy/models"
    
    os.makedirs(dest_dir, exist_ok=True)
    
    # Copy Routers (.pkl)
    patterns = ["router_*.pkl", "shard_sizes_*.json"]
    
    for pattern in patterns:
        files = glob.glob(os.path.join(src_dir, pattern))
        for f in files:
            shutil.copy(f, dest_dir)
            print(f"Copied {f} to {dest_dir}")

if __name__ == "__main__":
    copy_models()