"""Cache pinned inference files in the Space image before serving requests."""
from huggingface_hub import snapshot_download

from runtime import MODELS


if __name__ == "__main__":
    for model_id, spec in MODELS.items():
        snapshot_download(
            repo_id=model_id,
            revision=spec["revision"],
            allow_patterns=[
                "config.json", "generation_config.json", "model.safetensors",
                "tokenizer.json", "tokenizer_config.json", "tokenizer.model",
                "special_tokens_map.json", "added_tokens.json", "chat_template.jinja",
                "merges.txt", "vocab.json", "LICENSE",
            ],
        )