lalalalalalalalalala commited on
Commit
5ab6a59
·
verified ·
1 Parent(s): bab8ee6

Update run.py

Browse files
Files changed (1) hide show
  1. run.py +1 -1
run.py CHANGED
@@ -42,7 +42,7 @@ def fast_caption(sys_prompt, usr_prompt, temp, top_p, max_tokens, model, key, en
42
  token=video_hf_auth,
43
  )
44
  print(temp_parquet_file)
45
- parquet_file = pq.ParquetFile(parquet_path)
46
 
47
  for batch in parquet_file.iter_batches(batch_size=1):
48
  df = batch.to_pandas()
 
42
  token=video_hf_auth,
43
  )
44
  print(temp_parquet_file)
45
+ parquet_file = pq.ParquetFile(temp_parquet_file)
46
 
47
  for batch in parquet_file.iter_batches(batch_size=1):
48
  df = batch.to_pandas()