Update handler.py

handler.py

@@ -1,5 +1,5 @@
 from ultralyticsplus import YOLO
-from typing import Dict, Any, Optional, List
+from typing import List
 from sahi import ObjectPrediction
 
 
@@ -10,7 +10,7 @@ class EndpointHandler():
     def __init__(self):
         self.model = YOLO('ultralyticsplus/yolov8s')
 
-    def __call__(self, image, config: Optional[Dict[str, Any]] = None) -> List[ObjectPrediction]:
+    def __call__(self, image: str) -> List[ObjectPrediction]:
         """
        data args:
             image: image path to segment
@@ -21,8 +21,7 @@ class EndpointHandler():
       Return:
             object_predictions
         """
-        if config is None:
-            config = DEFAULT_CONFIG
+        config = DEFAULT_CONFIG
         # Set model parameters
         self.model.overrides['conf'] = config.get('conf')
         self.model.overrides['iou'] = config.get('iou')