diff --git a/Dockerfile b/Dockerfile index efd3b82..cc413bf 100644 --- a/Dockerfile +++ b/Dockerfile @@ -79,6 +79,9 @@ RUN if [ -n "$HF_TKN_FILE" ]; then \ EXPOSE 8000 +HEALTHCHECK --interval=30s --timeout=5s --start-period=120s --retries=3 \ + CMD python3 -c "import urllib.request; urllib.request.urlopen('http://localhost:8000/')" || exit 1 + ENTRYPOINT ["whisperlivekit-server", "--host", "0.0.0.0"] CMD ["--model", "medium"] diff --git a/Dockerfile.cpu b/Dockerfile.cpu index 5101fae..b5a3b2c 100644 --- a/Dockerfile.cpu +++ b/Dockerfile.cpu @@ -55,6 +55,9 @@ RUN if [ -n "$HF_TKN_FILE" ]; then \ # Expose port for the transcription server EXPOSE 8000 +HEALTHCHECK --interval=30s --timeout=5s --start-period=120s --retries=3 \ + CMD python3 -c "import urllib.request; urllib.request.urlopen('http://localhost:8000/')" || exit 1 + ENTRYPOINT ["whisperlivekit-server", "--host", "0.0.0.0"] # Default args - you might want to use a smaller model for CPU