funasr1.0

游雁

2023-12-27 ccb948895498b48e591f1f6a74cb62f4dcde8202

 funasr/models/neat_contextual_paraformer/model.py

@@ -53,9 +53,9 @@

from funasr.models.paraformer.model import Paraformer

from funasr.utils.register import register_class, registry_tables
from funasr.register import tables

@register_class("model_classes", "NeatContextualParaformer")
@tables.register("model_classes", "NeatContextualParaformer")
class NeatContextualParaformer(Paraformer):
   """
   Author: Speech Lab of DAMO Academy, Alibaba Group
@@ -417,7 +417,7 @@
               text = tokenizer.tokens2text(token)
               
               text_postprocessed, _ = postprocess_utils.sentence_postprocess(token)
               result_i = {"key": key[i], "token": token, "text": text, "text_postprocessed": text_postprocessed}
               result_i = {"key": key[i], "text": text_postprocessed}
               
               if ibest_writer is not None:
                  ibest_writer["token"][key[i]] = " ".join(token)