diff --git a/itn/chinese/inverse_normalizer.py b/itn/chinese/inverse_normalizer.py index 13aef9da..0f08b2c6 100644 --- a/itn/chinese/inverse_normalizer.py +++ b/itn/chinese/inverse_normalizer.py @@ -35,11 +35,13 @@ def __init__( self, cache_dir=None, overwrite_cache=False, + remove_interjections=True, enable_standalone_number=True, enable_0_to_9=False, enable_million=False, ): super().__init__(name="zh_inverse_normalizer", ordertype="itn") + self.remove_interjections = remove_interjections self.convert_number = enable_standalone_number self.enable_0_to_9 = enable_0_to_9 self.enable_million = enable_million @@ -78,6 +80,6 @@ def build_verbalizer(self): | LicensePlate().verbalizer | Whitelist().verbalizer ).optimize() - postprocessor = PostProcessor(remove_interjections=True).processor + postprocessor = PostProcessor(remove_interjections=self.remove_interjections).processor self.verbalizer = (verbalizer @ postprocessor).star