From e40b1e3e6b3f15c25bd1107fced0c46b70d3d0e5 Mon Sep 17 00:00:00 2001
From: Daniel Gatis <danielgatis@gmail.com>
Date: Sat, 21 Nov 2020 11:37:12 -0300
Subject: [PATCH] add dockerfile

---
 Dockerfile | 7 +++++++
 README.md  | 8 ++++++++
 2 files changed, 15 insertions(+)
 create mode 100644 Dockerfile

diff --git a/Dockerfile b/Dockerfile
new file mode 100644
index 0000000..85179d8
--- /dev/null
+++ b/Dockerfile
@@ -0,0 +1,7 @@
+FROM nvidia/cuda
+
+RUN apt-get update && apt-get install -y --no-install-recommends python3 python3-pip python3-dev llvm llvm-dev
+RUN pip3 install rembg
+
+ENTRYPOINT ["rembg"]
+CMD []
diff --git a/README.md b/README.md
index c341743..c39facb 100644
--- a/README.md
+++ b/README.md
@@ -99,6 +99,14 @@ Then run
     cat input.png | python app.py > out.png
 ```
 
+### Usage as a docker
+
+Just run
+
+```
+  curl -s http://input.png | docker run -i -v ~/.u2net:/root/.u2net danielgatis/rembg:latest > output.png
+```
+
 ### Advance usage
 
 Sometimes it is possible to achieve better results by turning on alpha matting. Example: