Update app.py

app.py

@@ -90,7 +90,7 @@ def get_line(prompt, inputs_len):
         clean_up_tokenization_spaces=True,
         return_full_text=False
     )
-    return remove_punctuation(output[0]['generated_text'])
+    return remove_punctuation(output[0]['generated_text']).strip()
 
 def get_rhyming_line(prompt, rhyming_word, inputs_len):
     output = gpt2_pipeline(
@@ -100,7 +100,7 @@ def get_rhyming_line(prompt, rhyming_word, inputs_len):
         clean_up_tokenization_spaces=True,
         return_full_text=False
     )
-    gpt2_sentence = remove_punctuation(output[0]['generated_text'])
+    gpt2_sentence = remove_punctuation(output[0]['generated_text']).strip()
     while len(gpt2_sentence) == 0:
         output = gpt2_pipeline(
             prompt + ".",
@@ -109,9 +109,8 @@ def get_rhyming_line(prompt, rhyming_word, inputs_len):
             clean_up_tokenization_spaces=True,
             return_full_text=False
         )
-        gpt2_sentence = remove_punctuation(output[0]['generated_text'])
+        gpt2_sentence = remove_punctuation(output[0]['generated_text']).strip()
     
-    gpt2_sentence = gpt2_sentence.replace("\n", "")
     print(f"\nGetting rhyming line starting with '{gpt2_sentence}' and ending with rhyming word '{rhyming_word}'")
     sentence = gpt2_sentence + " ___ ___ ___ " + rhyming_word
     print(f"Original Sentence: {sentence}")
@@ -136,7 +135,7 @@ def gpt2_summary(topic):
         clean_up_tokenization_spaces=True,
         return_full_text=False
     )
-    return remove_punctuation(output[0]['generated_text'])
+    return remove_punctuation(output[0]['generated_text']).strip()
     
 def generate(topic, wiki=True):
     if wiki: