deblan-mirror/buzz
1
0
Fork 0
mirror of https://github.com/chidiwilliams/buzz.git synced 2026-03-14 22:55:46 +01:00
Code Issues Projects Releases Packages Wiki Activity

Use block duration in recording

Browse source
This commit is contained in:
Chidi Williams 2022-09-27 01:03:13 +01:00
commit 5575380a7c
1 changed files with 7 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

14
transcriber.py
View file

@ -30,22 +30,22 @@ class Transcriber:
self.task = task
self.queue: queue.Queue[np.ndarray] = queue.Queue()
def start_recording(self, num_block=160, input_device_index: Optional[int] = None):
def start_recording(self, block_duration=10, input_device_index: Optional[int] = None):
logging.debug("Recording... language \"%s\", model \"%s\", task \"%s\", device \"%s\"" %
(self.language, self.model_name, self.task, input_device_index))
self.stream = sounddevice.InputStream(
samplerate=whisper.audio.SAMPLE_RATE, blocksize=whisper.audio.N_FRAMES*num_block,
device=input_device_index, dtype="float32", channels=1,
callback=self.stream_callback)
samplerate=whisper.audio.SAMPLE_RATE,
blocksize=block_duration * whisper.audio.SAMPLE_RATE,
device=input_device_index, dtype="float32",
channels=1, callback=self.stream_callback)
self.stream.start()
while True:
chunk = self.queue.get()
block = self.queue.get()
logging.debug(
'Processing next frame. Current queue size: %d' % self.queue.qsize())
result = self.model.transcribe(
audio=chunk, language=self.language, task=self.task)
audio=block, language=self.language, task=self.task)
logging.debug("Received next result: \"%s\"" % result["text"])
self.text_callback(result["text"])