Drazcat-AI commited on
Commit
f9df8b3
·
verified ·
1 Parent(s): 462fabe

Update handler.py

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