From 9f7b08e278bcf6fbf73c243f5befe4231509b219 Mon Sep 17 00:00:00 2001 From: Moritz Date: Wed, 14 Aug 2024 10:25:57 +0200 Subject: [PATCH] add OMP_NUM_THREADS --- docker-compose.yml | 3 ++- whisper.py | 3 +-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/docker-compose.yml b/docker-compose.yml index c35157c..c567ec9 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -7,6 +7,7 @@ services: - ASR_MODEL=large-v3 #- ASR_ENGINE=openai_whisper - ASR_ENGINE=faster_whisper + - OMP_NUM_THREADS=10 ports: - "127.0.0.1:9000:9000" volumes: @@ -14,7 +15,7 @@ services: - cache-pip:/root/.cache/pip - cache-poetry:/root/.cache/poetry - cache-whisper:/root/.cache/whisper - entrypoint: ["gunicorn", "--bind", "0.0.0.0:9000", "--workers", "10", "--timeout", "0", "app.webservice:app", "-k", "uvicorn.workers.UvicornWorker"] + #entrypoint: ["gunicorn", "--bind", "0.0.0.0:9000", "--workers", "10", "--timeout", "0", "app.webservice:app", "-k", "uvicorn.workers.UvicornWorker"] volumes: cache-pip: diff --git a/whisper.py b/whisper.py index 639355f..46838b9 100644 --- a/whisper.py +++ b/whisper.py @@ -19,12 +19,11 @@ def audio_callback_closure(client: AsyncClient): response = requests.post('http://127.0.0.1:9000/asr?encode=true&task=transcribe&word_timestamps=false&output=txt', files=files) print("Finished transcribing") room_id = room.room_id - sent_message = await client.room_send( + await client.room_send( room_id=room_id, message_type="m.room.message", content={"msgtype": "m.text", "body": response.content.decode()}, ) - print(sent_message) return audio_callback