Merge pull request #100 from Pbatch/pb_onnx

Add providers argument to support new ONNX versions
Pbatch · Dec 4, 2023 · a786beb · a786beb
2 parents 84b5e2b + 3d9fe32
commit a786beb
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/clashroyalebuildabot/state/onnx_detector.py b/clashroyalebuildabot/state/onnx_detector.py
@@ -6,7 +6,8 @@ class OnnxDetector:
     def __init__(self, model_path):
         self.model_path = model_path
 
-        self.sess = onnxruntime.InferenceSession(self.model_path)
+        self.sess = onnxruntime.InferenceSession(self.model_path,
+                                                 providers=['CPUExecutionProvider'])
         self.output_name = self.sess.get_outputs()[0].name
         self.input_name = self.sess.get_inputs()[0].name