Update Dockerfile

Dockerfile

@@ -62,6 +62,9 @@ RUN pip install --no-cache-dir --upgrade -r requirements.txt
 #COPY --chown=user teaching_arithmetic /teaching_arithmetic
 #COPY --chown=user . /app
 COPY . /app
+RUN mkdir ./out
+COPY ./out /out
+COPY ./out/addition_train /out/addition_train
 COPY ./train.sh /train.sh
 RUN chmod +x /train.sh
 # USER 1000