Spaces:
Runtime error
Runtime error
Update run.py
Browse files
run.py
CHANGED
|
@@ -19,7 +19,7 @@ def load_hf_dataset(dataset_path, auth_token):
|
|
| 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.TemporaryDirectory() as temp_dir:
|
| 22 |
-
temp_dir = '/opt'
|
| 23 |
csv_filename = os.path.join(temp_dir, 'caption.csv')
|
| 24 |
print(csv_filename)
|
| 25 |
with open(csv_filename, mode='w', newline='') as csv_file:
|
|
|
|
| 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.TemporaryDirectory() as temp_dir:
|
| 22 |
+
temp_dir = '/opt/run'
|
| 23 |
csv_filename = os.path.join(temp_dir, 'caption.csv')
|
| 24 |
print(csv_filename)
|
| 25 |
with open(csv_filename, mode='w', newline='') as csv_file:
|