Update run.py

run.py

@@ -19,9 +19,9 @@ def load_hf_dataset(dataset_path, auth_token):
 def fast_caption(sys_prompt, usr_prompt, temp, top_p, max_tokens, model, key, endpoint, video_src, video_hf, video_hf_auth, parquet_index, video_od, video_od_auth, video_gd, video_gd_auth, frame_format, frame_limit):
     progress_info = []
     with tempfile.TemporaryDirectory() as temp_dir:
-    # temp_dir = '/opt/run'
-    csv_filename = os.path.join(temp_dir, 'caption.csv')
-    print(csv_filename)
+        # temp_dir = '/opt/run'
+        csv_filename = os.path.join(temp_dir, 'caption.csv')
+        print(csv_filename)
         with open(csv_filename, mode='w', newline='') as csv_file:
             fieldnames = ['md5', 'caption']
             writer = csv.DictWriter(csv_file, fieldnames=fieldnames)