Jojohickman21 commited on
Commit
606af18
·
verified ·
1 Parent(s): b4f8247

Update inference.py

Browse files
Files changed (1) hide show
  1. inference.py +1 -1
inference.py CHANGED
@@ -17,7 +17,7 @@ class YOLOv10:
17
 
18
  def initialize_model(self, path):
19
  self.session = onnxruntime.InferenceSession(
20
- path, providers=onnxruntime.get_available_providers()
21
  )
22
  # Get model info
23
  self.get_input_details()
 
17
 
18
  def initialize_model(self, path):
19
  self.session = onnxruntime.InferenceSession(
20
+ path, providers=["CPUExecutionProvider"]
21
  )
22
  # Get model info
23
  self.get_input_details()