remove type hint

classification/inference_onnx.py

@@ -47,7 +47,7 @@ session = load_model(os.path.join(dirname, "./model/model_16.onnx"))
 print("model_path classification", os.path.join(dirname, "./model/model_16.onnx"))
 
 
-def prepare_input(image: NDArray[np.uint8]) -> NDArray[np.float16]:
+def prepare_input(image):
     """
     Prepare image input for model inference.
 
@@ -81,7 +81,7 @@ def softmax(x):
 classes = ["LEFT", "RIGHT", "STRAIGHT"]
 
 
-def inference(image: NDArray[np.uint8]) -> tuple[int, float, NDArray[np.float16]]:
+def inference(image):
     """
     Run inference on an image and return class prediction with probabilities.