YinuoGuo27 commited on
Commit
8c6551a
·
verified ·
1 Parent(s): 46f51d8

Update difpoint/src/models/predictor.py

Browse files
Files changed (1) hide show
  1. difpoint/src/models/predictor.py +2 -2
difpoint/src/models/predictor.py CHANGED
@@ -194,8 +194,8 @@ class OnnxRuntimePredictor:
194
  #self.inputs = self.onnx_model.get_inputs()
195
  #self.outputs = self.onnx_model.get_outputs()
196
  self.onnx_model = None
197
- self.inputs = None
198
- self.outputs = None
199
 
200
  def _load_model(self):
201
  """Lazy initialization of the ONNX model (only when needed)."""
 
194
  #self.inputs = self.onnx_model.get_inputs()
195
  #self.outputs = self.onnx_model.get_outputs()
196
  self.onnx_model = None
197
+ self.inputs = []
198
+ self.outputs = []
199
 
200
  def _load_model(self):
201
  """Lazy initialization of the ONNX model (only when needed)."""