PawanratRung commited on
Commit
068b839
·
verified ·
1 Parent(s): 6f7d99b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -156,7 +156,11 @@ def inference(image, model_name, mask_threshold, bbox_threshold):
156
  # Create an inference session.
157
  ort_session = onnxruntime.InferenceSession(
158
  path_onnx,
159
- providers=["CPUExecutionProvider"],
 
 
 
 
160
  sess_options=sess_options,
161
  )
162
 
 
156
  # Create an inference session.
157
  ort_session = onnxruntime.InferenceSession(
158
  path_onnx,
159
+ providers=["CUDAExecutionProvider", "CPUExecutionProvider"],
160
+ sess_options=sess_options,
161
+ )
162
+
163
+
164
  sess_options=sess_options,
165
  )
166