daihui.zhang
commited on
Commit
·
62d476f
1
Parent(s):
471affe
fix init language setting bug
Browse files
transcribe/pipelines/pipe_whisper.py
CHANGED
|
@@ -30,4 +30,4 @@ class WhisperPipe(BasePipe):
|
|
| 30 |
for char in bytearray_chars.decode('utf-8', errors='replace'):
|
| 31 |
if unicodedata.category(char) != 'Cc': # 不可打印字符的分类为 'Cc'
|
| 32 |
printable.append(char)
|
| 33 |
-
return ''.join(printable)
|
|
|
|
| 30 |
for char in bytearray_chars.decode('utf-8', errors='replace'):
|
| 31 |
if unicodedata.category(char) != 'Cc': # 不可打印字符的分类为 'Cc'
|
| 32 |
printable.append(char)
|
| 33 |
+
return ''.join(printable).strip()
|
transcribe/whisper_llm_serve.py
CHANGED
|
@@ -40,7 +40,7 @@ class PyWhiperCppServe(ServeClientBase):
|
|
| 40 |
self.run_in_thread(self.speech_to_text)
|
| 41 |
self.run_in_thread(self.get_frame_from_queue)
|
| 42 |
|
| 43 |
-
self.text_sep = ""
|
| 44 |
|
| 45 |
def run_in_thread(self, func):
|
| 46 |
t = threading.Thread(target=func)
|
|
@@ -57,6 +57,7 @@ class PyWhiperCppServe(ServeClientBase):
|
|
| 57 |
def set_lang(self, src_lang, dst_lang):
|
| 58 |
self.language = src_lang
|
| 59 |
self.dst_lang = dst_lang
|
|
|
|
| 60 |
|
| 61 |
def add_frames(self, frame_np):
|
| 62 |
self._frame_queue.put(frame_np)
|
|
|
|
| 40 |
self.run_in_thread(self.speech_to_text)
|
| 41 |
self.run_in_thread(self.get_frame_from_queue)
|
| 42 |
|
| 43 |
+
self.text_sep = ""
|
| 44 |
|
| 45 |
def run_in_thread(self, func):
|
| 46 |
t = threading.Thread(target=func)
|
|
|
|
| 57 |
def set_lang(self, src_lang, dst_lang):
|
| 58 |
self.language = src_lang
|
| 59 |
self.dst_lang = dst_lang
|
| 60 |
+
self.text_sep = "" if self.language == "zh" else " "
|
| 61 |
|
| 62 |
def add_frames(self, frame_np):
|
| 63 |
self._frame_queue.put(frame_np)
|