rizavelioglu commited on
Commit
75b360c
1 Parent(s): 35ed39f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -156,7 +156,7 @@ 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