skumar1998
/

yolotesting

Model card Files Files and versions

skumar1998 commited on Dec 27, 2023

Commit

83d24c3

·

1 Parent(s): c9cf887

Update handler.py

Files changed (1) hide show

handler.py +2 -3

handler.py CHANGED Viewed

@@ -15,9 +15,8 @@ class EndpointHandler():
         current_directory = os.getcwd()
         logging.info(f'current dir: {current_directory}')
         logging.info(f'all files: {os.listdir(path)}')
-        # model_path = "yolov8m_detect_usdl.pt"
-        # self.model = YOLO(model_path)
-        # print('Yolo model loaded successfully')
     def __call__(self, data: Dict) -> Dict:

         current_directory = os.getcwd()
         logging.info(f'current dir: {current_directory}')
         logging.info(f'all files: {os.listdir(path)}')
+        self.model = YOLO(os.path.join(path, 'yolov8m_detect_usdl.pt'))
+        print('Yolo model loaded successfully')
     def __call__(self, data: Dict) -> Dict: