update docker compose

Author: micaelakaplan

label_studio_ml/examples/segment_anything_2_image/Dockerfile

@@ -21,7 +21,7 @@ ENV PYTHONUNBUFFERED=1 \
     PORT=9090 \
     WORKERS=2 \
     THREADS=4 \
-    CUDA_HOME=/usr/local/cuda \
+    CUDA_HOME=/usr/local/cuda
 
 # this image is already built with cuda 12.1.1, so we don't need to install it again
 # RUN conda install -c "nvidia/label/cuda-12.1.1" cuda -y 
@@ -39,8 +39,8 @@ RUN --mount=type=cache,target=${PIP_CACHE_DIR},sharing=locked \
     pip3 install -r requirements.txt
 
 # install segment-anything-2
-RUN cd ~ && git clone --depth 1 --branch main --single-branch https://github.com/facebookresearch/segment-anything-2.git
-WORKDIR /segment-anything-2
+RUN cd / && git clone --depth 1 --branch main --single-branch https://github.com/facebookresearch/sam2.git
+WORKDIR /sam2
 RUN --mount=type=cache,target=${PIP_CACHE_DIR},sharing=locked \
     pip3 install -e .
 RUN cd checkpoints && ./download_ckpts.sh