from pathlib import Path
from semantic_model import SemanticColorizer, load_semantic, save_semantic
def load_model(source, revision=None, device="cpu"):
    from huggingface_hub import snapshot_download
    path=Path(source)
    if not path.is_dir(): path=Path(snapshot_download(source,revision=revision,allow_patterns=["config.json","model.safetensors"]))
    return load_semantic(path,device)
save_model=save_semantic