Update Dockerfile and README

Author: yummybian

Dockerfile

@@ -1,27 +1,22 @@
 FROM yummybian/docker-darknet:latest
 
-ENV KAI-REPO kai-repo
 
 # Working directory
 WORKDIR /kai-service
 
-go get https://github.com/ZanLabs/kai.git
-
 # Download And Install Kai
-RUN git clone https://github.com/ZanLabs/kai.git KAI-REPO && \
-    cd KAI-REPO && \
-    go install && \
-    mv kai .. && \
-    mv cfg .. && \
-    mv data .. && \ 
-    mv config.yaml .. && \
-    cd .. && \
-    rm -rf KAI-REPO 
+RUN go get -u github.com/ZanLabs/kai && \
+    go install github.com/ZanLabs/kai && \
+    cp /go/bin/kai ./kai
+
+# Copy configurations
+RUN cp -Ra /go/src/github.com/ZanLabs/kai/cfg ./ && \
+    cp -Ra /go/src/github.com/ZanLabs/kai/data ./ && \
+    cp /go/src/github.com/ZanLabs/kai/config.yaml ./config.yaml
 
 # Download weights
 RUN curl -O http://pjreddie.com/media/files/yolo.weights >/dev/null 2>&1 
 
-CMD [ "./kai" ]
-
-
+EXPOSE 8000
 
+ENTRYPOINT [ "./kai" ]

README.md

@@ -12,6 +12,19 @@
 - support S3 download and upload
 - support Ftp download and upload
 
+## Precondition
+- nvidia-docker 2.0
+- CUDA
+- cudnn 
+
+## Docker
+```bash
+git pull yummybian/kai
+sudo docker run --runtime nvidia -it --rm -p 8000:8000 -v /path/to/config.yaml:/kai-service/config.yaml yummybian/kai bash
+or
+sudo docker run --runtime nvidia -d --name kai -p 8000:8000 -v /path/to/config.yaml:/kai-service/config.yaml yummybian/kai 
+```
+
 ## Setting Up
 
 First make sure you have [go-yolo](https://github.com/ZanLabs/go-yolo) installed on your machine.