Install rembg also for GPU

PPCBee · Feb 21, 2024 · 181f865 · 181f865
1 parent 9783899
commit 181f865
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Dockerfile b/Dockerfile
@@ -5,7 +5,7 @@ WORKDIR /rembg
 COPY . .
 
 RUN python -m pip install -r requirements.txt
-RUN python -m pip install ".[cli]"
+RUN python -m pip install ".[gpu,cli]"
 RUN python -c 'from rembg.bg import download_models; download_models()'
 
 EXPOSE 5000