hnluo
2023-05-19 c4e37cb6c179d35030c211004e6875a016eb7972
funasr/bin/asr_infer.py
@@ -9,6 +9,7 @@
import time
import copy
import os
import re
import codecs
import tempfile
import requests
@@ -828,9 +829,16 @@
                # Change integer-ids to tokens
                token = self.converter.ids2tokens(token_int)
                token = " ".join(token)
                results.append(token)
                postprocessed_result = ""
                for item in token:
                    if item.endswith('@@'):
                        postprocessed_result += item[:-2]
                    elif re.match('^[a-zA-Z]+$', item):
                        postprocessed_result += item + " "
                    else:
                        postprocessed_result += item
                results.append(postprocessed_result)
        # assert check_return_type(results)
        return results