[refactor][abaomxomtieu]: put model out scrop of inference function

src/inference/segment_inference.py

@@ -2,11 +2,13 @@ import onnxruntime as ort
 from src.utils.utils_segment import preprocess, postprocess
 import numpy as np
 
+model_path = "./src/model/segment.onnx"
+model = ort.InferenceSession(
+    model_path,
+)
 
-def inference(image: np.array, model_path, threshold_confidence=0.5, threshold_iou=0.7):
-    model = ort.InferenceSession(
-        model_path,
-    )
+
+def inference(image: np.array, threshold_confidence=0.5, threshold_iou=0.7):
     input = preprocess(image)
     outputs = postprocess(
         model.run(None, {"images": input}),
@@ -16,8 +18,3 @@ def inference(image: np.array, model_path, threshold_confidence=0.5, threshold_i
 
     return outputs
 
-
-if __name__ == "__main__":
-    model_path = "../model/segment.onnx"
-    image_path = "../../test.jpg"
-    print(inference(image_path, model_path))