diff --git a/whisper_streaming/cli.py b/whisper_streaming/cli.py index 33b53db..4cd0343 100644 --- a/whisper_streaming/cli.py +++ b/whisper_streaming/cli.py @@ -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}") diff --git a/whisper_streaming/transcriber.py b/whisper_streaming/transcriber.py index 7037ffb..2b0d248 100644 --- a/whisper_streaming/transcriber.py +++ b/whisper_streaming/transcriber.py @@ -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(