增加挂载目录

docker/build.sh

@@ -10,7 +10,7 @@ docker save model_watermark_generate -o model_watermark_generate.tar
 docker load -i model_watermark_generate.tar
 
 # 运行容器
-docker run -d -p 8888:5000 --name watermark_container model_watermark_generate -v /home/:/home/ai/
+docker run -d -p 8888:5000 --name watermark_container -v /home/:/home/ai/ model_watermark_generate
 
 # 进入容器
 docker exec -it watermark_container /bin/bash