Remove starting newlines on generated text

minimaxir · Jun 16, 2019 · ae3436d · ae3436d
1 parent 6951425
commit ae3436d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/gpt_2_simple/gpt_2.py b/gpt_2_simple/gpt_2.py
@@ -429,10 +429,11 @@ def generate(sess,
                 trunc_text = re.search(pattern, gen_text, re.S)
                 if trunc_text:
                     gen_text = trunc_text.group(1)
+            gen_text = gen_text.lstrip('\n')
             if destination_path:
                 f.write("{}\n{}".format(gen_text, sample_delim))
             if not return_as_list and not destination_path:
-                print("{}\n{}".format(gen_text, sample_delim))
+                print("{}\n{}".format(gen_text, sample_delim), end='')
             gen_texts.append(gen_text)
 
     if destination_path: