Update docker files

.dockerignore

@@ -2,6 +2,7 @@
 !.streamlit/
 !figures/
 !utils/
+!checkpoint/
 !app.py
 !index.html
 !style.css

Dockerfile

@@ -2,34 +2,29 @@ FROM nvidia/cuda:11.7.1-runtime-ubuntu22.04
 
 RUN apt update && apt upgrade -y && \
     apt install -y bash \
-                   build-essential \
-                   git \
-                   curl \
-                   ca-certificates \
-                   python3-dev \
-                   python3-pip && \
+    build-essential \
+    git \
+    curl \
+    ca-certificates \
+    python3-dev \
+    python3-pip && \
     rm -rf /var/lib/apt/lists
 
-RUN apt install software-properties-common && add-apt-repository ppa:deadsnakes/ppa && \
-    apt update && apt install python3.9 && \
-    rm -rf /var/lib/apt/lists
-RUN rm -f /usr/bin/python && ln -s /usr/bin/python /usr/bin/python3.9
-
-RUN python3.9 -m pip install --no-cache-dir --upgrade pip && \
-    python33.9 -m pip install --no-cache-dir --extra-index-url https://download.pytorch.org/whl/cu113 \
+RUN python3.10 -m pip install --no-cache-dir --upgrade pip && \
+    python3.10 -m pip install --no-cache-dir --extra-index-url https://download.pytorch.org/whl/cu117 \
     torch \
     torchvision \
     torchaudio
 
-WORKDIR /code
+RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install ffmpeg libsm6 libxext6  -y && rm -rf /var/lib/apt/lists
 
-COPY ./requirements.txt /code/requirements.txt
+RUN python3.10 -m pip install --no-cache-dir git+https://github.com/facebookresearch/segment-anything.git
 
-RUN python3.9 -m pip install --no-cache-dir --upgrade -r /code/requirements.txt
+WORKDIR /code
 
-RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install ffmpeg libsm6 libxext6  -y && rm -rf /var/lib/apt/lists
+COPY ./requirements.txt /code/requirements.txt
 
-RUN python3.9 -m pip install --no-cache-dir git+https://github.com/bowang-lab/MedSAM.git
+RUN python3.10 -m pip install --no-cache-dir --upgrade -r /code/requirements.txt
 
 # Set up a new user named "user" with user ID 1000
 RUN useradd -m -u 1000 user
@@ -46,4 +41,4 @@ WORKDIR $HOME/app
 # Copy the current directory contents into the container at $HOME/app setting the owner to the user
 COPY --chown=user . $HOME/app
 
-CMD ["python3.9", "-m", "streamlit", "run", "app.py"]
+CMD ["python3.10", "-m", "streamlit", "run", "app.py"]