panda1835 commited on
Commit
1e9b835
1 Parent(s): 8a8a3ca

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +2 -1
handler.py CHANGED
@@ -7,7 +7,8 @@ class EndpointHandler():
7
  # pseudo:
8
  # self.model= load_model(path)
9
  # yolov8_model_name = 'yolov8_2023-07-19_yolov8m.pt'
10
- self.model = YOLO(path)
 
11
 
12
  def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
13
  """
 
7
  # pseudo:
8
  # self.model= load_model(path)
9
  # yolov8_model_name = 'yolov8_2023-07-19_yolov8m.pt'
10
+ print(path)
11
+ self.model = YOLO('./yolov8_2023-07-19_yolov8m.pt')
12
 
13
  def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
14
  """