Spaces:
Runtime error
Runtime error
Update run.py
Browse files
run.py
CHANGED
@@ -18,8 +18,8 @@ def load_hf_dataset(dataset_path, auth_token):
|
|
18 |
|
19 |
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):
|
20 |
progress_info = []
|
21 |
-
|
22 |
-
|
23 |
fieldnames = ['md5', 'caption']
|
24 |
writer = csv.DictWriter(csv_file, fieldnames=fieldnames)
|
25 |
writer.writeheader()
|
|
|
18 |
|
19 |
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):
|
20 |
progress_info = []
|
21 |
+
with tempfile.NamedTemporaryFile(mode='w', delete=False, newline='') as csv_file:
|
22 |
+
csv_filename = csv_file.name
|
23 |
fieldnames = ['md5', 'caption']
|
24 |
writer = csv.DictWriter(csv_file, fieldnames=fieldnames)
|
25 |
writer.writeheader()
|