utils/model_loader.py

from huggingface_hub import hf_hub_download

def load_model():
    model_path = hf_hub_download(
        repo_id="your-username/your-repo",
        filename="best.pt",
        revision="main"
    )
    return YOLO(model_path)