lalalalalalalalalala commited on
Commit
e6b969d
·
verified ·
1 Parent(s): 6a77982

Update run.py

Browse files
Files changed (1) hide show
  1. run.py +1 -1
run.py CHANGED
@@ -69,7 +69,7 @@ def fast_caption(sys_prompt, usr_prompt, temp, top_p, max_tokens, model, key, en
69
  writer.writerow({'md5': md5, 'caption': caption})
70
  # writer.writerow({'md5': md5, 'caption': 'caption'})
71
  # progress_info.append(f"Processed video with MD5: {md5}")
72
- if ind == 10:
73
  return csv_filename, "\n".join(progress_info), None
74
  # return csv_filename, "\n".join(progress_info), None
75
  else:
 
69
  writer.writerow({'md5': md5, 'caption': caption})
70
  # writer.writerow({'md5': md5, 'caption': 'caption'})
71
  # progress_info.append(f"Processed video with MD5: {md5}")
72
+ if ind == 86:
73
  return csv_filename, "\n".join(progress_info), None
74
  # return csv_filename, "\n".join(progress_info), None
75
  else: