diff --git a/.gitignore b/.gitignore
index bee8a64..dadcbe4 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1 +1,3 @@
 __pycache__
+*.pth
+*.pkl
diff --git a/Dockerfile b/Dockerfile
index 222483c..9fb890e 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -15,13 +15,18 @@ ENV \
 
 RUN \
   apt-get update && \
-  apt-get install -y --no-install-recommends tini build-essential gfortran libatlas-base-dev && \
+  apt-get install -y --no-install-recommends tini build-essential gfortran libatlas-base-dev wget && \
   pip install "poetry==1.1.13"
 
 COPY pyproject.toml poetry.lock ./
 RUN \
   python -m poetry install --no-dev && \
   rm -rf /root/.cache/pypoetry/artifacts /root/.cache/pypoetry/cache
+
+RUN \
+  mkdir models && \
+  wget https://github.com/danbooru/autotagger/releases/download/2022.06.20-233624-utc/model.pth -O models/model.pth
+
 COPY . .
 
 EXPOSE 5000
diff --git a/models/.gitkeep b/models/.gitkeep
new file mode 100644
index 0000000..e69de29