Spaces:
Runtime error
Runtime error
Update run.py
Browse files
run.py
CHANGED
@@ -23,7 +23,7 @@ def load_hf_dataset(dataset_path, auth_token):
|
|
23 |
video_paths = dataset
|
24 |
return video_paths
|
25 |
|
26 |
-
def fast_caption(sys_prompt, usr_prompt, temp, top_p, max_tokens, model, key, endpoint,
|
27 |
progress_info = []
|
28 |
with tempfile.TemporaryDirectory() as temp_dir:
|
29 |
# temp_dir = '/opt/run'
|
|
|
23 |
video_paths = dataset
|
24 |
return video_paths
|
25 |
|
26 |
+
def fast_caption(sys_prompt, usr_prompt, temp, top_p, max_tokens, model, key, endpoint, video_hf, video_hf_auth, parquet_index, video_od, video_od_auth, video_gd, video_gd_auth, frame_format, frame_limit):
|
27 |
progress_info = []
|
28 |
with tempfile.TemporaryDirectory() as temp_dir:
|
29 |
# temp_dir = '/opt/run'
|