Update presentation_assistant/presentation_assistant.py

presentation_assistant/presentation_assistant.py

@@ -76,9 +76,9 @@ 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,
-                    "max_new_tokens": 500}})
-    # reply = output[0]['generated_text'][len("You are a kind helpful PPT designer. "+input_prompt):]
-    reply = output[0]['generated_text']
+                    "max_new_tokens": 200}})
+    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')
@@ -140,8 +140,8 @@ 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,
-                    "max_new_tokens": 500}})
-    # reply = output[0]['generated_text'][len("You are a kind helpful PPT Assistant."+input_prompt):]
-    reply = output[0]['generated_text']
+                    "max_new_tokens": 200}})
+    reply = output[0]['generated_text'][len("You are a kind helpful PPT Assistant."+input_prompt):]
+    # reply = output[0]['generated_text']
 
     return reply