File size: 307 Bytes
11a04cf
da6a2fa
 
 
11a04cf
da6a2fa
d07fe7a
da6a2fa
 
 
 
1
2
3
4
5
6
7
8
9
10
11
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)