Selaa lähdekoodia

fix vad realtime space bug

mengzhe.cmz 2 vuotta sitten
vanhempi
sitoutus
3d7c0c5d2d

+ 1 - 1
funasr/bin/punctuation_infer_vadrealtime.py

@@ -70,7 +70,7 @@ class Text2Punc:
         else:
             precache = ""
             cache = []
-        data = {"text": precache + text}
+        data = {"text": precache + " " + text}
         result = self.preprocessor(data=data, uid="12938712838719")
         split_text = self.preprocessor.pop_split_text_data(result)
         mini_sentences = split_to_mini_sentence(split_text, split_size)

+ 1 - 1
funasr/runtime/python/onnxruntime/funasr_onnx/punc_bin.py

@@ -148,7 +148,7 @@ class CT_Transformer_VadRealtime(CT_Transformer):
         else:
             precache = ""
             cache = []
-        full_text = precache + text
+        full_text = precache + " " + text
         split_text = code_mix_split_words(full_text)
         split_text_id = self.converter.tokens2ids(split_text)
         mini_sentences = split_to_mini_sentence(split_text, split_size)