diff --git a/daq/src/aaredaq/mlbox.py b/daq/src/aaredaq/mlbox.py index a850af50..c94911fa 100644 --- a/daq/src/aaredaq/mlbox.py +++ b/daq/src/aaredaq/mlbox.py @@ -1,9 +1,12 @@ +from pathlib import Path from typing import Tuple from ultralytics import YOLO + class MlBox: - def __init__(self, model_path: str = "/home/leonarski_f/mxlibs3/src/aaredaq/best.pt"): + def __init__(self, model_path: str = "best.pt"): + model_path = str(Path(__file__).parent / model_path) self.__model = YOLO(model_path) @staticmethod