whitespace

app.py

@@ -57,6 +57,8 @@ def continue_story(prompt):
     print("generated text was",output[0]['generated_text'])
 
     output_tmp = output[0]['generated_text']
+    #strip whitespace
+    output_tmp = output_tmp.strip()
     #truncate response at first newline
     if "\n" in output_tmp:
         idx = output_tmp.find('\n')