Update run.py

run.py

@@ -37,12 +37,13 @@ def fast_caption(sys_prompt, usr_prompt, temp, top_p, max_tokens, model, key, en
         all_captions = []
         with tempfile.NamedTemporaryFile(mode='w+t', delete=True) as temp_parquet_file:
             temp_parquet_file = hf_hub_download(
-                repo_id="OpenVideo/pexels-raw",
+                repo_id=video_hf,
                 filename="data/“ + str(number).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)
             
             for batch in parquet_file.iter_batches(batch_size=1):
@@ -53,6 +54,7 @@ def fast_caption(sys_prompt, usr_prompt, temp, top_p, max_tokens, model, key, en
                 with tempfile.NamedTemporaryFile(mode='w+t', delete=True) as temp_video_file:
                     temp_video_file.write(video)
                     video_path = temp_video_file.name
+                    print(video_path)
 
                     processor = VideoProcessor(frame_format=frame_format, frame_limit=frame_limit)
                     frames = processor._decode(video_path)