diff --git a/Dockerfile b/Dockerfile
index 3eb3a52..8905bbe 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -6,7 +6,7 @@ RUN pip install --upgrade pip
 
 COPY . .
 
-RUN python -m pip install ".[cli]"
+RUN python -m pip install ".[cpu,cli]"
 RUN rembg d
 
 EXPOSE 7000