Add docker

README.md

@@ -105,6 +105,18 @@ streamlit run app.py
 ```
 and then open the browser and go to the address `http://localhost:8501`.
 
+## How to load Trained Model from Hugging Face
+The trained model is also uploaded to [hugging face](https://huggingface.co/afshin-dini/Egg-Detection) from which one can use it as following:
+```shell
+from huggingface_hub import hf_hub_download
+from ultralytics import YOLO
+
+model_path = hf_hub_download(repo_id="afshin-dini/Egg-Detection", filename="model/egg_detector.pt")
+model = YOLO(model_path)
+result = model("path/to/image")
+```
+Then, the uploaded model can be used for different purposes.
+
 ## How to Develop
 Do the following only once after creating your project:
 - Init the git repo with `git init`.
@@ -121,7 +133,31 @@ Then create a branch with `git checkout -b BRANCH_NAME` for further developments
 - For applying changes use `pre-commit run --all-files`.
 
 ## Docker Container
-Under development.
+To run the docker with ssh, do the following first and then based on your need select ,test, development, or production containers:
+```shell
+export DOCKER_BUILDKIT=1
+export DOCKER_SSHAGENT="-v $SSH_AUTH_SOCK:$SSH_AUTH_SOCK -e SSH_AUTH_SOCK"
+```
+### Test Container
+This container is used for testing purposes while it runs the test
+```shell
+docker build --progress plain --ssh default --target test -t package_detection_docker:test .
+docker run -it --rm -v "$(pwd):/app" $(echo $DOCKER_SSHAGENT) package_detection_docker:test
+```
+
+### Development Container
+This container can be used for development purposes:
+```shell
+docker build --progress plain --ssh default --target development -t package_detection_docker:development .
+docker run -it --rm -v "$(pwd):/app" -v /tmp:/tmp $(echo $DOCKER_SSHAGENT) package_detection_docker:development
+```
+
+### Production Container
+This container can be used for production purposes:
+```shell
+docker build --progress plain --ssh default --target production -t package_detection_docker:production .
+docker run -it --rm -v "$(pwd):/app" -v /tmp:/tmp $(echo $DOCKER_SSHAGENT) package_detection_docker:production deep_package_detection -vv infer --model_path src/deep_package_detection/model/package_detection.pt --data_path ./tests/test_data --result_path ./results
+```
 
 ## License
 This repo is licensed under the Attribution–NonCommercial 4.0 International License (CC BY-NC 4.0).