Update run.py

run.py

@@ -38,7 +38,13 @@ def fast_caption(sys_prompt, usr_prompt, temp, top_p, max_tokens, model, key, en
                 progress_info.append('Begin processing Hugging Face dataset.')
                 os.environ["TOKIO_WORKER_THREADS"] = "8"
                 os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1"
-                pqfile = hf_hub_download(repo_id=video_hf, filename='data/' + str(parquet_index).zfill(6) + '.parquet', repo_type="dataset", local_dir="/dev/shm", token=video_hf_auth)
+                pqfile = hf_hub_download(
+                    repo_id=video_hf,
+                    filename='data/' + str(parquet_index).zfill(6) + '.parquet',
+                    repo_type="dataset",
+                    local_dir="/dev/shm",
+                    token=video_hf_auth,
+                )
 
                 pf = pq.ParquetFile(pqfile)
                 for batch in pf.iter_batches(1):