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)
|