diff --git a/notebooks/onnx_model_example.ipynb b/notebooks/onnx_model_example.ipynb index ca49c3571..3a109096d 100644 --- a/notebooks/onnx_model_example.ipynb +++ b/notebooks/onnx_model_example.ipynb @@ -135,6 +135,8 @@ "import numpy as np\n", "import cv2\n", "import matplotlib.pyplot as plt\n", + "import sys\n", + "sys.path.append(\"..\")\n", "from segment_anything import sam_model_registry, SamPredictor\n", "from segment_anything.utils.onnx import SamOnnxModel\n", "\n",