mirrors/whispering
1
0
Fork 0
mirror of https://github.com/shirayu/whispering.git synced 2024-05-20 09:18:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add debug logs

Browse source
This commit is contained in:
Yuta Hayashibe 2022-09-24 10:09:16 +09:00
parent 85f9b8fc13
commit 12cd9a4f42
2 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
whisper_streaming/cli.py
View file

@ -42,7 +42,7 @@ def transcribe_from_mic(
):
idx: int = 0
while True:
logger.debug(f"Segment: {idx}")
logger.debug(f"Segment #: {idx}, Queue: {q.qsize()}")
segment = q.get()
for chunk in wsp.transcribe(segment=segment):
print(f"{chunk.start:.2f}->{chunk.end:.2f}\t{chunk.text}")

3
whisper_streaming/transcriber.py
View file

@ -90,6 +90,7 @@ class WhisperStreamingTranscriber:
) -> List[DecodingResult]:
assert len(self.config.temperatures) >= 1
t = self.config.temperatures[0]
logger.debug(f"temperature: {t}")
_decode_options1: DecodingOptions = self._get_decoding_options(
t=t,
@ -108,6 +109,7 @@ class WhisperStreamingTranscriber:
and result.avg_logprob < self.config.logprob_threshold
for result in results
]
logger.debug(f"Fall back with temperature: {t}, {needs_fallback}")
if any(needs_fallback):
_decode_options2: DecodingOptions = self._get_decoding_options(
t=t,
@ -123,6 +125,7 @@ class WhisperStreamingTranscriber:
np.nonzero(needs_fallback)[0]
):
results[original_index] = retries[retry_index]
logger.debug(f"# of results: {len(results)}")
return results
def _get_chunk(