Add files via upload
This commit is contained in:
1 parent
b2f9adad66
commit
bb09b75d3b
1 file changed
+1
-1
+1
-1
@@ -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
|
||||
|
||||
|
||||
Reference in new issue
Block a user