diff --git a/README.md b/README.md index dfdf64f..d1876e5 100644 --- a/README.md +++ b/README.md @@ -2,4 +2,7 @@ python3 -m venv .venv source .venv/bin/activate pip3 install -r requirements.txt + +docker compose up +python3 whisper.py ``` diff --git a/docker-compose.yml b/docker-compose.yml new file mode 100644 index 0000000..713b810 --- /dev/null +++ b/docker-compose.yml @@ -0,0 +1,22 @@ +version: "3.4" + +services: + whisper-asr-webservice: + image: onerahmet/openai-whisper-asr-webservice:latest + environment: + - ASR_MODEL=large-v3 + #- ASR_ENGINE=openai_whisper + - ASR_ENGINE=faster_whisper + ports: + - "127.0.0.1:9000:9000" + volumes: + # ./app:/app/app + - cache-pip:/root/.cache/pip + - cache-poetry:/root/.cache/poetry + - cache-whisper:/root/.cache/whisper + +volumes: + cache-pip: + cache-poetry: + cache-whisper: + cache-faster-whisper: diff --git a/whisper.py b/whisper.py index 3aef652..01e3684 100644 --- a/whisper.py +++ b/whisper.py @@ -17,7 +17,6 @@ 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) room_id = room.room_id await client.room_send( - # Watch out! If you join an old room you'll see lots of old messages room_id=room_id, message_type="m.room.message", content={"msgtype": "m.text", "body": response.content.decode()}, @@ -39,6 +38,7 @@ async def main() -> None: await client.sync() for room in client.invited_rooms: await client.join(room) + # Watch out! If you join an old room you'll see lots of old messages await client.sync_forever(timeout=30000) # milliseconds asyncio.run(main())