Update Dockerfiles and dependencies for multi-architecture support with optional extras for Torch CPU and GPU env

Author: ahmetoner

Dockerfile

@@ -14,13 +14,13 @@ ENV PATH="${PATH}:${POETRY_VENV}/bin"
 
 WORKDIR /app
 
-COPY . /app
+COPY . .
 COPY --from=ffmpeg /usr/local/bin/ffmpeg /usr/local/bin/ffmpeg
 COPY --from=swagger-ui /usr/share/nginx/html/swagger-ui.css swagger-ui-assets/swagger-ui.css
 COPY --from=swagger-ui /usr/share/nginx/html/swagger-ui-bundle.js swagger-ui-assets/swagger-ui-bundle.js
 
 RUN poetry config virtualenvs.in-project true
-RUN poetry install
+RUN poetry install --extras cpu
 
 EXPOSE 9000
 

Dockerfile.gpu

@@ -30,19 +30,14 @@ ENV PATH="${PATH}:${POETRY_VENV}/bin"
 
 WORKDIR /app
 
-COPY poetry.lock pyproject.toml ./
-
-RUN poetry config virtualenvs.in-project true
-RUN poetry install --no-root
-
 COPY . .
 COPY --from=ffmpeg /usr/local/bin/ffmpeg /usr/local/bin/ffmpeg
 COPY --from=swagger-ui /usr/share/nginx/html/swagger-ui.css swagger-ui-assets/swagger-ui.css
 COPY --from=swagger-ui /usr/share/nginx/html/swagger-ui-bundle.js swagger-ui-assets/swagger-ui-bundle.js
 
-RUN poetry install
-RUN $POETRY_VENV/bin/pip install torch==2.6.0+cu126 torchaudio==2.6.0+cu126 --index-url https://download.pytorch.org/whl/cu126
+RUN poetry config virtualenvs.in-project true
+RUN poetry install --extras cuda
 
 EXPOSE 9000
 
-CMD whisper-asr-webservice
+ENTRYPOINT ["whisper-asr-webservice"]

README.md

@@ -75,11 +75,14 @@ For complete documentation, visit:
 ## Development
 
 ```shell
-# Install poetry
+# Install poetry v2.X
 pip3 install poetry
 
-# Install dependencies
-poetry install
+# Install dependencies for cpu
+poetry install --extras cpu
+
+# Install dependencies for cuda
+poetry install --extras cuda
 
 # Run service
 poetry run whisper-asr-webservice --host 0.0.0.0 --port 9000

docs/build.md

@@ -1,17 +1,23 @@
 ## Development Environment
 
-Install poetry with following command:
+Install poetry v2.X with following command:
 
 ```shell
 pip3 install poetry
 ```
 
 ### Installation
 
-Install packages:
+Install dependencies for cpu
 
 ```shell
-poetry install
+poetry install --extras cpu
+```
+
+Install dependencies for cuda
+
+```shell
+poetry install --extras cuda
 ```
 
 !!! Note