ED-Demo commited on
Commit
81454e6
·
verified ·
1 Parent(s): b10e919

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -17,7 +17,7 @@ class EndpointHandler:
17
  self.device = "cpu"
18
 
19
  # Load YOLO model
20
- self.model = YOLO(path)
21
  print(f"✅ Model loaded on {self.device}")
22
 
23
  def __call__(self, data):
 
17
  self.device = "cpu"
18
 
19
  # Load YOLO model
20
+ self.model = YOLO(model_path)
21
  print(f"✅ Model loaded on {self.device}")
22
 
23
  def __call__(self, data):