diff --git a/docker/Dockerfile b/docker/Dockerfile
index 5089ea7..eece0fd 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -1,6 +1,10 @@
 # Use an official CUDA runtime with Ubuntu as a parent image
 FROM nvidia/cuda:12.4.1-runtime-ubuntu22.04
 
+ARG GIT_REPO=https://github.com/theroyallab/tabbyAPI
+ARG DO_PULL=true
+ENV DO_PULL $DO_PULL
+
 # Set the working directory in the container
 WORKDIR /app
 
@@ -14,6 +18,14 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
     git \
     && rm -rf /var/lib/apt/lists/*
 
+# Update repo
+RUN if [ ${DO_PULL} ]; then \
+    git init && \
+    git remote add origin $GIT_REPO && \
+    git fetch origin && \
+    git pull origin main && \
+    echo "Pull finished"; fi
+
 # Upgrade pip
 RUN pip3 install --no-cache-dir --upgrade pip
 
diff --git a/docker/docker-compose.yml b/docker/docker-compose.yml
index d50682e..fd6634c 100644
--- a/docker/docker-compose.yml
+++ b/docker/docker-compose.yml
@@ -4,6 +4,8 @@ services:
     build:
       context: ..
       dockerfile: ./docker/Dockerfile
+      args:
+        - DO_PULL=true
     ports:
       - "5000:5000"
     environment: