Merge pull request #199 from beeldengeluid/198-add-standard-log-forma…

…t-to-whisper_api added time to log messages
beeldengeluid · Jan 16, 2025 · 5dc6720 · 5dc6720
2 parents 7a8b68d + e59e34b
commit 5dc6720
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 2 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -9,7 +9,7 @@ ENV POETRY_NO_INTERACTION=1 \
 
 COPY ./poetry.lock ./poetry.lock
 COPY ./pyproject.toml ./pyproject.toml
-RUN poetry self add poetry-plugin-export
+RUN poetry self add poetry-plugin-export==1.8.0
 RUN poetry export --format requirements.txt --output requirements.txt
 
 FROM nvidia/cuda:12.3.2-cudnn9-runtime-ubuntu22.04

diff --git a/main.py b/main.py
@@ -15,8 +15,11 @@ def run_api(port: int):
     from whisper_api import api
     import uvicorn
 
+    log_config = uvicorn.config.LOGGING_CONFIG
+    log_config["formatters"]["default"]["fmt"] = LOG_FORMAT
+
     logger.info("Running Whisper as a service")
-    uvicorn.run(api, port=port, host="0.0.0.0")
+    uvicorn.run(api, port=port, host="0.0.0.0", log_config=log_config)
 
 
 def run_job(input_uri: str, output_uri: str):

diff --git a/whisper_api.py b/whisper_api.py
@@ -12,6 +12,7 @@
     w_model,
 )
 
+
 logger = logging.getLogger(__name__)
 api = FastAPI()
-Original file line number
+Diff line change
@@ Expand Up / @@ -12,6 +12,7 @@ @@
         w_model,
     )
     logger = logging.getLogger(__name__)
     api = FastAPI()
@@ Expand Down @@