From 0bc9cb23729eccd5077b1a0b417eed9f8f21414e Mon Sep 17 00:00:00 2001
From: evazion <noizave@gmail.com>
Date: Mon, 20 Jun 2022 23:44:03 -0500
Subject: [PATCH] docker: add model to image.

---
 .gitignore      | 2 ++
 Dockerfile      | 7 ++++++-
 models/.gitkeep | 0
 3 files changed, 8 insertions(+), 1 deletion(-)
 create mode 100644 models/.gitkeep

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