akhaliq HF staff commited on
Commit
1c181c5
·
1 Parent(s): a302618

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
  sess = rt.InferenceSession('efficientnet-lite4-11.onnx')
51
 
52
  def inference(img):
 
47
  img = img[top:bottom, left:right]
48
  return img
49
 
50
+ options = ort.SessionOptions()
51
+ options.intra_op_num_threads = 1
52
+ options.inter_op_num_threads = 1
53
  sess = rt.InferenceSession('efficientnet-lite4-11.onnx')
54
 
55
  def inference(img):