LPX55
commited on
Commit
·
0adef02
1
Parent(s):
ceff40b
fix: add missing import for postprocess_onnx_output in model_loader
Browse files- utils/model_loader.py +1 -0
utils/model_loader.py
CHANGED
|
@@ -1,3 +1,4 @@
|
|
|
|
|
| 1 |
# Add missing import for infer_onnx_model
|
| 2 |
from utils.onnx_helpers import infer_onnx_model
|
| 3 |
# Add missing import for preprocess_onnx_input
|
|
|
|
| 1 |
+
from utils.onnx_helpers import postprocess_onnx_output
|
| 2 |
# Add missing import for infer_onnx_model
|
| 3 |
from utils.onnx_helpers import infer_onnx_model
|
| 4 |
# Add missing import for preprocess_onnx_input
|