Fixed model download + add to image

.dockerignore

@@ -3,8 +3,6 @@ build/
 dist/
 livermask.egg-info/
 *.DS_Store
-*.h5
-*.npz
 *.nii
 *__pycache__/
 .idea/

Dockerfile

@@ -22,10 +22,6 @@ WORKDIR /code
 RUN apt-get update -y
 RUN apt install git --fix-missing -y
 
-# Download pretrained parenchyma model
-RUN wget "https://github.com/andreped/livermask/releases/download/trained-models-v1/model.h5"
-COPY ./model.h5 /code/model.h5
-
 # install dependencies
 COPY ./demo/requirements.txt /code/demo/requirements.txt
 RUN python3.7 -m pip install --no-cache-dir --upgrade -r /code/demo/requirements.txt
@@ -33,6 +29,13 @@ RUN python3.7 -m pip install --no-cache-dir --upgrade -r /code/demo/requirements
 # resolve issue with tf==2.4 and gradio dependency collision issue
 RUN python3.7 -m pip install --force-reinstall typing_extensions==4.0.0
 
+# Install wget
+RUN apt install wget -y
+
+# Download pretrained parenchyma model
+RUN wget "https://github.com/andreped/livermask/releases/download/trained-models-v1/model.h5"
+#COPY --chown=user ./model.h5 /code/model.h5
+
 # Set up a new user named "user" with user ID 1000
 RUN useradd -m -u 1000 user