LPX55 commited on
Commit
3232315
·
1 Parent(s): eeb602e

fix: add missing import for infer_onnx_model in model_loader

Browse files
Files changed (1) hide show
  1. utils/model_loader.py +2 -0
utils/model_loader.py CHANGED
@@ -1,3 +1,5 @@
 
 
1
  # Add missing import for preprocess_onnx_input
2
  from utils.onnx_helpers import preprocess_onnx_input
3
  """
 
1
+ # Add missing import for infer_onnx_model
2
+ from app import infer_onnx_model
3
  # Add missing import for preprocess_onnx_input
4
  from utils.onnx_helpers import preprocess_onnx_input
5
  """