Update run.py

run.py

@@ -36,12 +36,12 @@ def fast_caption(sys_prompt, usr_prompt, temp, top_p, max_tokens, model, key, en
         # Process all videos in the dataset
         all_captions = []
         with tempfile.NamedTemporaryFile(mode='w+t', delete=True) as temp_parquet_file:
-            temp_parquet_file = hf_hub_download(
+            temp_parquet_file.write(hf_hub_download(
                 repo_id=video_hf,
                 filename='data/' + str(parquet_index).zfill(6) + '.parquet',
                 repo_type="dataset",
                 token=video_hf_auth,
-            )
+            ))
             parquet_path = temp_parquet_file.name
             print(parquet_path)
             parquet_file = pq.ParquetFile(parquet_path)