akhaliq HF Staff commited on
Commit
580867f
·
1 Parent(s): 3f07d4e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -47,6 +47,9 @@ def center_crop(img, out_height, out_width):
47
  img = img[top:bottom, left:right]
48
  return img
49
 
 
 
 
50
 
51
  sess = rt.InferenceSession('inception-v1-12.onnx')
52
 
 
47
  img = img[top:bottom, left:right]
48
  return img
49
 
50
+ options = rt.SessionOptions()
51
+ options.intra_op_num_threads = 1
52
+ options.inter_op_num_threads = 1
53
 
54
  sess = rt.InferenceSession('inception-v1-12.onnx')
55