Update presentation_assistant/presentation_assistant.py

presentation_assistant/presentation_assistant.py

@@ -78,7 +78,7 @@ def text2ppt(token_key, input_prompt, input_theme):
                     "return_full_text": False,
                     "max_new_tokens": 200}})
     # reply = output[0]['generated_text'][len("You are a kind helpful PPT designer. "+input_prompt):]
-    reply = output[0]['generated_text']
+    reply = output[0]['generated_text'].strip()
     print(reply)
     md_text = reply[4:] if reply[:3] == "---" else reply
     md_text_list = md_text.split('\n')
@@ -147,6 +147,6 @@ def ppt2script(token_key, input_file, input_type):
                     "return_full_text": False,
                     "max_new_tokens": 200}})
     # reply = output[0]['generated_text'][len("You are a kind helpful PPT Assistant."+input_prompt):]
-    reply = output[0]['generated_text']
+    reply = output[0]['generated_text'].strip()
 
     return reply