Drazcat-AI commited on
Commit
dc18c2d
·
verified ·
1 Parent(s): 3a0d088

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +2 -1
handler.py CHANGED
@@ -13,7 +13,8 @@ class EndpointHandler():
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()]
 
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
+ results = self.model(image_path)
18
  predictions = []
19
  for box in results[0].boxes:
20
  class_id = results[0].names[box.cls[0].item()]