change init file

StarlightSearch · Nov 11, 2024 · e45b111 · e45b111
1 parent 8b28553
commit e45b111
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/python/python/embed_anything/__init__.py b/python/python/embed_anything/__init__.py
@@ -125,9 +125,12 @@
 import glob
 
 path = os.path.dirname(onnxruntime.__file__) + "/capi/"
-dylib_path = glob.glob(os.path.join(path, "libonnxruntime.so*"))
 
-if os.environ.get("ORT_DYLIB_PATH") is None:
+if path is None:
+    print("onnxruntime is not installed. Install it using `pip install onnxruntime`")
+
+else:
+    dylib_path = glob.glob(os.path.join(path, "libonnxruntime.so*"))
     os.environ["ORT_DYLIB_PATH"] = dylib_path[0]
 
 __doc__ = _embed_anything.__doc__