Update presentation_assistant/presentation_assistant.py

presentation_assistant/presentation_assistant.py

@@ -76,22 +76,16 @@ def text2ppt(token_key, input_prompt, input_theme):
     output = query({"inputs": "You are a kind helpful PPT designer. "+input_prompt,     
                     "parameters": {
                     "return_full_text": False,
-                    "max_new_tokens": 200}})
-    # reply = output[0]['generated_text'][len("You are a kind helpful PPT designer. "+input_prompt):]
+                    "max_new_tokens": 1000}})
+    
     reply = output[0]['generated_text']
-    print(reply)
     
     md_text = reply[4:] if reply[:3] == "---" else reply
     md_text_list = md_text.split('\n')
-    print(md_text_list)
 
     f = open("text2ppt_input.md", 'w')
     for i in range(0, len(md_text_list)):
-        # data = ""
-        # if md_text_list[i] and "<!--" in md_text_list[i]:
-        #     data = "---" + "\n"
         data = md_text_list[i].strip() + "\n"
-        # print(data)
         f.write(data)
     f.close()
 
@@ -146,8 +140,7 @@ def ppt2script(token_key, input_file, input_type):
     output = query({"inputs": "You are a kind helpful PPT Assistant."+input_prompt,     
                     "parameters": {
                     "return_full_text": False,
-                    "max_new_tokens": 200}})
-    # reply = output[0]['generated_text'][len("You are a kind helpful PPT Assistant."+input_prompt):]
+                    "max_new_tokens": 1000}})
     reply = output[0]['generated_text']
 
     return reply