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

Update run.py

Browse files
Files changed (1) hide show
  1. run.py +2 -2
run.py CHANGED
@@ -68,8 +68,8 @@ def fast_caption(sys_prompt, usr_prompt, temp, top_p, max_tokens, model, key, en
68
  caption = api.get_caption(sys_prompt, usr_prompt, base64_list)
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:
 
68
  caption = api.get_caption(sys_prompt, usr_prompt, base64_list)
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: