daihui.zhang commited on
Commit
ded2334
·
1 Parent(s): df55474

add logging format

Browse files
Files changed (1) hide show
  1. transcribe/whisper_llm_serve.py +4 -2
transcribe/whisper_llm_serve.py CHANGED
@@ -264,8 +264,9 @@ class PyWhiperCppServe(ServeClientBase):
264
  transcribe_fut = self._pool.submit(
265
  PywhisperInference.transcribe, audio_buffer.tobytes(), self.language)
266
 
267
- log_block("Whisper transcrible time", f"{(time.perf_counter() - start_time):.3f}", "s")
268
  segments = transcribe_fut.result()
 
 
269
  return segments
270
 
271
  def translate_text(self, text):
@@ -277,8 +278,9 @@ class PyWhiperCppServe(ServeClientBase):
277
  translate_fut = self._pool.submit(
278
  PywhisperInference.translate, text, self.language, self.dst_lang)
279
 
 
280
  log_block("LLM translate time", f"{(time.perf_counter() - start_time):.3f}", "s")
281
- return translate_fut.result()
282
 
283
  def _segments_split(self, segments, audio_buffer: np.ndarray):
284
  """根据左边第一个标点符号来将序列拆分成 观察段 和 剩余部分"""
 
264
  transcribe_fut = self._pool.submit(
265
  PywhisperInference.transcribe, audio_buffer.tobytes(), self.language)
266
 
 
267
  segments = transcribe_fut.result()
268
+ log_block("Whisper transcrible time", f"{(time.perf_counter() - start_time):.3f}", "s")
269
+
270
  return segments
271
 
272
  def translate_text(self, text):
 
278
  translate_fut = self._pool.submit(
279
  PywhisperInference.translate, text, self.language, self.dst_lang)
280
 
281
+ ret = translate_fut.result()
282
  log_block("LLM translate time", f"{(time.perf_counter() - start_time):.3f}", "s")
283
+ return ret
284
 
285
  def _segments_split(self, segments, audio_buffer: np.ndarray):
286
  """根据左边第一个标点符号来将序列拆分成 观察段 和 剩余部分"""