Spaces:
Sleeping
Sleeping
Update run.py
Browse files
run.py
CHANGED
@@ -38,7 +38,7 @@ def fast_caption(sys_prompt, usr_prompt, temp, top_p, max_tokens, model, key, en
|
|
38 |
with tempfile.NamedTemporaryFile(mode='w+t', delete=True) as temp_parquet_file:
|
39 |
temp_parquet_file = hf_hub_download(
|
40 |
repo_id=video_hf,
|
41 |
-
filename='data/' + str(
|
42 |
repo_type="dataset",
|
43 |
token=video_hf_auth,
|
44 |
)
|
|
|
38 |
with tempfile.NamedTemporaryFile(mode='w+t', delete=True) as temp_parquet_file:
|
39 |
temp_parquet_file = hf_hub_download(
|
40 |
repo_id=video_hf,
|
41 |
+
filename='data/' + str(parquet_index).zfill(6) + '.parquet',
|
42 |
repo_type="dataset",
|
43 |
token=video_hf_auth,
|
44 |
)
|