iarbel
/

yolov8s_test

Object Detection

ultralyticsplus

Inference Endpoints

Model card Files Files and versions Community

iarbel commited on Jan 4

Commit

3a43c10

•

1 Parent(s): 5bb5008

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -6,7 +6,7 @@ DEFAULT_CONFIG = {'conf': 0.25, 'iou': 0.45, 'agnostic_nms': False, 'max_det': 1
 BOX_KEYS = ['xmin', 'ymin', 'xmax', 'ymax']
 class EndpointHandler():
-    def __init__(self):
         self.model = YOLO('ultralyticsplus/yolov8s')
     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:

 BOX_KEYS = ['xmin', 'ymin', 'xmax', 'ymax']
 class EndpointHandler():
+    def __init__(self, path=""):
         self.model = YOLO('ultralyticsplus/yolov8s')
     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]: