Update Dockerfile

docker/Dockerfile

@@ -1,6 +1,6 @@
 FROM nvidia/cuda:12.1.0-devel-ubuntu22.04
 ENV DEBIAN_FRONTEND=noninteractive
-RUN apt update && apt install python3-pip -y
+RUN apt update && apt install python3-pip wget -y
 RUN apt-get install -y git
 
 RUN pip3 install torch==2.3.1 torchvision==0.18.1 --index-url https://download.pytorch.org/whl/cu121
@@ -26,7 +26,6 @@ RUN pip3 install mock==5.1.0
 RUN mkdir -p sam2.1_weights
 
 # Download SAM 2.1 checkpoints
-RUN apt-get install -y wget
 RUN wget -O /sam2_weights/sam2.1_hiera_tiny.pt "https://dl.fbaipublicfiles.com/segment_anything_2/092824/sam2.1_hiera_tiny.pt"
 RUN wget -O /sam2_weights/sam2.1_hiera_small.pt "https://dl.fbaipublicfiles.com/segment_anything_2/092824/sam2.1_hiera_small.pt"
 RUN wget -O /sam2_weights/sam2.1_hiera_base_plus.pt "https://dl.fbaipublicfiles.com/segment_anything_2/092824/sam2.1_hiera_base_plus.pt"