haoneng.lhn
2023-05-04 297fafd674715bcd849557616bd22b6791460f31
update streaming paraformer text process
1个文件已修改
13 ■■■■ 已修改文件
funasr/bin/asr_inference_paraformer_streaming.py 13 ●●●● 补丁 | 查看 | 原始文档 | blame | 历史
funasr/bin/asr_inference_paraformer_streaming.py
@@ -239,7 +239,7 @@
                        feats_len = torch.tensor([feats_chunk2.shape[1]])
                        results_chunk2 = self.infer(feats_chunk2, feats_len, cache)
                        return ["".join(results_chunk1 + results_chunk2)]
                        return [" ".join(results_chunk1 + results_chunk2)]
                results = self.infer(feats, feats_len, cache)
@@ -299,12 +299,13 @@
                # Change integer-ids to tokens
                token = self.converter.ids2tokens(token_int)
                token = " ".join(token)
                if self.tokenizer is not None:
                    text = self.tokenizer.tokens2text(token)
                else:
                    text = None
                results.append(text)
                #if self.tokenizer is not None:
                #    text = self.tokenizer.tokens2text(token)
                #else:
                #    text = None
                results.append(token)
        # assert check_return_type(results)
        return results