Update generate.py

2019-10-24 09:25:16 +08:00
parent 2ffcc14a08
commit 4e56c1b3b6
1 changed files with 3 additions and 1 deletions
--- a/generate.py
+++ b/generate.py
@@ -198,7 +198,9 @@ def main():
                for i, item in enumerate(text):
                    if item == '[MASK]':
                        text[i] = ''
-                    if item == '[CLS]' or item == '[SEP]':
+                    elif item == '[CLS]':
+                        text[i] = '\n\n'
+                    elif item == '[SEP]':
                        text[i] = '\n'
                info = "=" * 40 + " SAMPLE " + str(generated) + " " + "=" * 40 + "\n"
                print(info)