Update presentation_assistant/presentation_assistant.py

presentation_assistant/presentation_assistant.py

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