from ultralytics import YOLO
from huggingface_hub import hf_hub_download

def load_model():
    # Download model from Hugging Face Hub (if hosted)
    model_path = hf_hub_download(
        repo_id="MobvxtR/mbModel/models"
        filename="best.pt",
        revision="main"
    )
    return YOLO(model_path)