Skip to content

Commit d29351a

Browse files
duj12dujing
andauthored
add remove_interjections option for chinese itn. (#324)
Co-authored-by: dujing <dujing@xmov.ai>
1 parent 64761cb commit d29351a

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

itn/chinese/inverse_normalizer.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,11 +35,13 @@ def __init__(
3535
self,
3636
cache_dir=None,
3737
overwrite_cache=False,
38+
remove_interjections=True,
3839
enable_standalone_number=True,
3940
enable_0_to_9=False,
4041
enable_million=False,
4142
):
4243
super().__init__(name="zh_inverse_normalizer", ordertype="itn")
44+
self.remove_interjections = remove_interjections
4345
self.convert_number = enable_standalone_number
4446
self.enable_0_to_9 = enable_0_to_9
4547
self.enable_million = enable_million
@@ -78,6 +80,6 @@ def build_verbalizer(self):
7880
| LicensePlate().verbalizer
7981
| Whitelist().verbalizer
8082
).optimize()
81-
postprocessor = PostProcessor(remove_interjections=True).processor
83+
postprocessor = PostProcessor(remove_interjections=self.remove_interjections).processor
8284

8385
self.verbalizer = (verbalizer @ postprocessor).star

0 commit comments

Comments
 (0)