Spaces:
Sleeping
Sleeping
Update difpoint/src/models/predictor.py
Browse files
difpoint/src/models/predictor.py
CHANGED
@@ -173,7 +173,7 @@ class OnnxRuntimePredictor:
|
|
173 |
"""
|
174 |
OnnxRuntime Prediction
|
175 |
"""
|
176 |
-
|
177 |
def __init__(self, **kwargs):
|
178 |
model_path = kwargs.get("model_path", "") # 用模型路径区分是否是一样的实例
|
179 |
assert os.path.exists(model_path), "model path must exist!"
|
|
|
173 |
"""
|
174 |
OnnxRuntime Prediction
|
175 |
"""
|
176 |
+
|
177 |
def __init__(self, **kwargs):
|
178 |
model_path = kwargs.get("model_path", "") # 用模型路径区分是否是一样的实例
|
179 |
assert os.path.exists(model_path), "model path must exist!"
|