Update presentation_assistant/presentation_assistant.py

presentation_assistant/presentation_assistant.py

@@ -75,10 +75,10 @@ def generate_text2ppt_input_prompt(input_type, input_value, input_pages):
 def text2ppt(token_key, input_prompt, input_theme):
     output = query({"inputs": "You are a kind helpful PPT designer. "+input_prompt,     
                     "parameters": {
-                    "return_full_text": True,
+                    "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'][len("You are a kind helpful PPT designer. "+input_prompt):]
+    reply = output[0]['generated_text']
     print(reply)
     md_text = reply[4:] if reply[:3] == "---" else reply
     md_text_list = md_text.split('\n')
@@ -139,9 +139,9 @@ def ppt2script(token_key, input_file, input_type):
     
     output = query({"inputs": "You are a kind helpful PPT Assistant."+input_prompt,     
                     "parameters": {
-                    "return_full_text": True,
+                    "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'][len("You are a kind helpful PPT Assistant."+input_prompt):]
+    reply = output[0]['generated_text']
 
     return reply