daihui.zhang
commited on
Commit
·
ded2334
1
Parent(s):
df55474
add logging format
Browse files
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
|
| 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 |
"""根据左边第一个标点符号来将序列拆分成 观察段 和 剩余部分"""
|