diff --git a/lib/python/bailoclient/resources/templates/onxx.py b/lib/python/bailoclient/resources/templates/onnx.py similarity index 99% rename from lib/python/bailoclient/resources/templates/onxx.py rename to lib/python/bailoclient/resources/templates/onnx.py index 245a27af1..312bc8067 100644 --- a/lib/python/bailoclient/resources/templates/onxx.py +++ b/lib/python/bailoclient/resources/templates/onnx.py @@ -6,7 +6,7 @@ class OnnxModel(BaseModel): - def __init__(self, path, providers=None): + def __init__(self, path): import onnxruntime import os import numpy as np