small fix

Former-commit-id: e95cbc81622f36854334feb89608e69614fa0972

pipeline.py

@@ -16,14 +16,16 @@ if args.link is None and args.local_path is None:
     print("need video source")
     exit()
 
-openai.api_key = "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"  
+# openai.api_key = "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"  
+openai.api_key = "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"
 # openai.api_key = os.getenv("OPENAI_API_KEY")
 
 DOWNLOAD_PATH = args.download
 RESULT_PATH = args.result
 VIDEO_NAME = args.video_name
 n_threshold = 5000
-model_name = "text-davinci-003" # replace this to our own fintune model
+# model_name = "text-davinci-003" # replace this to our own fintune model
+model_name = "gpt-3.5-turbo"
 
 
 # get source audio
@@ -57,7 +59,7 @@ with open("{}/{}_en.txt".format(RESULT_PATH, VIDEO_NAME), 'r') as f:
     script_split = script_en.split('.')
 
 # Split the video script by sentences and create chunks within the token limit
-n_threshold = 4096  # Token limit for the GPT-3 model
+n_threshold = 2048  # Token limit for the GPT-3 model
 script_split = script_en.split('.')
 
 script_arr = []
@@ -78,7 +80,7 @@ for s in script_arr:
         model=model_name,
         prompt=prompt,
         temperature=0.1,
-        max_tokens=2000,
+        max_tokens=2048,
         top_p=1.0,
         frequency_penalty=0.0,
         presence_penalty=0.0