Merge pull request martinakaduc#6 from sangnguyenimp21/fix/blank_space

fix: remove blank space before char
nhatkhangcs · Jun 21, 2023 · f929e81 · f929e81
2 parents b66955a + 2c8119c
commit f929e81
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/pipeline/translation.py b/pipeline/translation.py
@@ -22,6 +22,10 @@ def post_process(text):
             else:
                 continue
         output = " ".join(output)
+        special_chars = [',', '.', ':', '?', '!']
+        for char in special_chars:
+            if char in output:
+                output = output.replace(' '+char, char)
         output = output[0].capitalize() + output[1:]
         return output