jpterry commited on
Commit
6e2e849
·
1 Parent(s): 979296b

adding options to runtime initialization

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -46,8 +46,9 @@ def load_model(model: str, activation: bool=True):
46
 
47
  print(model)
48
 
49
- ort_session = ort.InferenceSession(model_path + '%s.onnx' % (model))
50
- ort_session.intra_op_num_threads = 0
 
51
 
52
  return ort_session
53
 
 
46
 
47
  print(model)
48
 
49
+ sess_options = ort.SessionOptions()
50
+ sess_options.intra_op_num_threads = 0
51
+ ort_session = ort.InferenceSession(model_path + '%s.onnx' % (model), sess_options)
52
 
53
  return ort_session
54