local-it/whisper-bot
0
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

add more worker

Browse source
This commit is contained in:
Moritz 2024-08-14 10:05:33 +02:00
parent 6aec5226ba
commit cb45e37b68
2 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
docker-compose.yml
View file

@ -14,6 +14,7 @@ services:
- cache-pip:/root/.cache/pip - cache-pip:/root/.cache/pip
- cache-poetry:/root/.cache/poetry - cache-poetry:/root/.cache/poetry
- cache-whisper:/root/.cache/whisper - cache-whisper:/root/.cache/whisper
entrypoint: ["gunicorn", "--bind", "0.0.0.0:9000", "--workers", "10", "--timeout", "0", "app.webservice:app", "-k", "uvicorn.workers.UvicornWorker"]
volumes: volumes:
cache-pip: cache-pip:

4
whisper.py
View file

@ -17,12 +17,14 @@ def audio_callback_closure(client: AsyncClient):
# file.write(audio.body) # file.write(audio.body)
print("Transcribe audio") print("Transcribe audio")
response = requests.post('http://127.0.0.1:9000/asr?encode=true&task=transcribe&word_timestamps=false&output=txt', files=files) 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 room_id = room.room_id
await client.room_send( sent_message = await client.room_send(
room_id=room_id, room_id=room_id,
message_type="m.room.message", message_type="m.room.message",
content={"msgtype": "m.text", "body": response.content.decode()}, content={"msgtype": "m.text", "body": response.content.decode()},
) )
print(sent_message)
return audio_callback return audio_callback