diff --git a/PYAS_Engine.py b/PYAS_Engine.py index c9c23e7..0313ddb 100644 --- a/PYAS_Engine.py +++ b/PYAS_Engine.py @@ -122,7 +122,7 @@ class model_scanner: def load_file(self, file): if file.lower().endswith('.onnx'): try: - self.model = onnxruntime.InferenceSession(file, providers=['CPUExecutionProvider']) + self.model = onnxruntime.InferenceSession(file) except Exception: return False