fix segment anything model

Author: micaelakaplan

label_studio_ml/examples/segment_anything_model/Dockerfile

@@ -23,17 +23,21 @@ RUN --mount=type=cache,target="/var/cache/apt",sharing=locked \
     apt-get update; \
     apt-get upgrade -y; \
     apt install --no-install-recommends -y  \
-        wget git libopencv-dev python3-opencv cmake protobuf-compiler; \
+        wget git libopencv-dev cmake protobuf-compiler; \
     apt-get autoremove -y
 
 # Copy and run the model download script
 COPY download_models.sh .
 RUN bash /app/download_models.sh
 
+# Install numpy first to avoid conflicts with system numpy
+RUN --mount=type=cache,target=${PIP_CACHE_DIR},sharing=locked \
+    pip install --upgrade pip && \
+    pip install "numpy>=2,<2.3.0"
+
 # install base requirements
 COPY requirements-base.txt .
 RUN --mount=type=cache,target=${PIP_CACHE_DIR},sharing=locked \
-    pip install --upgrade pip && \
     pip install -r requirements-base.txt
 
 # install custom requirements

label_studio_ml/examples/segment_anything_model/requirements.txt

@@ -1,7 +1,8 @@
+numpy>=2,<2.3.0
 label_studio_converter
-opencv-python
+opencv-python-headless>=4.12.0,<5.0.0
 onnxruntime==1.15.1
-onnx
+onnx==1.12.0
 torch==2.0.1
 torchvision==0.15.2
 gunicorn==22.0.0
@@ -11,5 +12,3 @@ timm==0.4.12
 segment_anything @ git+https://github.com/facebookresearch/segment-anything.git
 mobile-sam @ git+https://github.com/ChaoningZhang/MobileSAM.git
 label-studio-ml @ git+https://github.com/heartexlabs/label-studio-ml-backend.git
-
-numpy<2