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 = "" if self.language == "zh" else " "
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)