Update handler.py

handler.py

@@ -9,7 +9,8 @@ class EndpointHandler():
         self.model = YOLO(hf_hub_download(repo_id="Drazcat-AI/flejes", filename="yolov8_flejes/runs/detect/train/weights/best.pt", local_files_only=True))
     
     def predict_objects(self, image_path):
-        results = self.model(image_path, imgsz=1280)
+        #results = self.model(image_path, imgsz=1280)
+        results = self.model(image_path, imgsz=800)
         predictions = []
         for box in results[0].boxes:
             class_id = results[0].names[box.cls[0].item()]