DatSplit commited on
Commit
6b42faf
·
verified ·
1 Parent(s): 3f46474

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -142,7 +142,7 @@ def inference(image_path, model_name, bbox_threshold):
142
  sess_options.graph_optimization_level = onnxruntime.GraphOptimizationLevel.ORT_DISABLE_ALL
143
  ort_session = onnxruntime.InferenceSession(
144
  model_path,
145
- providers=["CUDAExecutionProvider", "CPUExecutionProvider"],
146
  sess_options=sess_options
147
  )
148
 
 
142
  sess_options.graph_optimization_level = onnxruntime.GraphOptimizationLevel.ORT_DISABLE_ALL
143
  ort_session = onnxruntime.InferenceSession(
144
  model_path,
145
+ providers=["CPUExecutionProvider"],
146
  sess_options=sess_options
147
  )
148