Drazcat-AI commited on
Commit
847c565
·
verified ·
1 Parent(s): 39c617a

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -9,7 +9,7 @@ class EndpointHandler():
9
  self.model = YOLO(hf_hub_download(repo_id="Drazcat-AI/flejes", filename="yolov8_flejes/runs/detect/train/weights/best.pt", local_files_only=True))
10
 
11
  def predict_objects(self, image_path):
12
- results = self.model(image_path, imgsz=800)
13
  predictions = []
14
  for box in results[0].boxes:
15
  class_id = results[0].names[box.cls[0].item()]
 
9
  self.model = YOLO(hf_hub_download(repo_id="Drazcat-AI/flejes", filename="yolov8_flejes/runs/detect/train/weights/best.pt", local_files_only=True))
10
 
11
  def predict_objects(self, image_path):
12
+ results = self.model(image_path, imgsz=1280)
13
  predictions = []
14
  for box in results[0].boxes:
15
  class_id = results[0].names[box.cls[0].item()]