From f455a0573656fcf95da54a538341f1672c556126 Mon Sep 17 00:00:00 2001
From: ko1N <ko1N1337@gmail.com>
Date: Sun, 17 Jan 2021 16:07:54 +0100
Subject: [PATCH] Dockerfile now uses the same train_log folder as the scripts

---
 README.md         | 2 +-
 docker/Dockerfile | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/README.md b/README.md
index 6e377f1..65e008e 100644
--- a/README.md
+++ b/README.md
@@ -81,7 +81,7 @@ ffmpeg -r 10 -f image2 -i output/img%d.png -s 448x256 -vf "split[s0][s1];[s0]pal
 ```
 
 ### Run in docker
-Place the pre-trained models in the `./docker/pretrained_models directory`
+Place the pre-trained models in `train_log/\*.pkl` (as above)
 
 Building the container:
 ```
diff --git a/docker/Dockerfile b/docker/Dockerfile
index eca0589..801dbb7 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -15,7 +15,7 @@ ADD docker/inference_video /usr/local/bin/inference_video
 RUN chmod +x /usr/local/bin/inference_video
 
 # add pre-trained models
-COPY docker/pretrained_models /rife/train_log
+COPY train_log /rife/train_log
 
 WORKDIR /host
 ENTRYPOINT ["/bin/bash"]