Update run.py

run.py

@@ -28,9 +28,8 @@ def fast_caption(sys_prompt, usr_prompt, temp, top_p, max_tokens, model, key, en
     progress_info = []
     with tempfile.TemporaryDirectory() as temp_dir:
         # temp_dir = '/opt/run'
-        csv_filename = os.path.join('/dev/shm', str(parquet_index).zfill(6) + '_gpt4o_caption.csv')
-        # csv_filename = '/dev/shm/caption.csv'
-        print(csv_filename)
+        # csv_filename = os.path.join('/dev/shm', str(parquet_index).zfill(6) + '_gpt4o_caption.csv')
+        csv_filename = '/dev/shm/caption.csv'
         with open(csv_filename, mode='w', newline='') as csv_file:
             fieldnames = ['md5', 'caption']
             writer = csv.DictWriter(csv_file, fieldnames=fieldnames)
@@ -70,10 +69,10 @@ def fast_caption(sys_prompt, usr_prompt, temp, top_p, max_tokens, model, key, en
                             # writer.writerow({'md5': md5, 'caption': 'caption'})
                             progress_info.append(f"Processed video with MD5: {md5}")
                             i += 1
-                        if i == 1:
-                            break
-                        #return csv_filename, "\n".join(progress_info), None
-                return csv_filename, "\n".join(progress_info), None
+                        # if i == 1:
+                        #     break
+                        return csv_filename, "\n".join(progress_info), None
+                # return csv_filename, "\n".join(progress_info), None
             else:
                 return "", "No video source selected.", None