lalalalalalalalalala commited on
Commit
c91a2e2
·
verified ·
1 Parent(s): 1f3c35f

Update run.py

Browse files
Files changed (1) hide show
  1. run.py +2 -2
run.py CHANGED
@@ -69,9 +69,9 @@ 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 == 86:
73
  return csv_filename, "\n".join(progress_info), None
74
- return csv_filename, "\n".join(progress_info), None
75
  else:
76
  return "", "No video source selected.", None
77
 
 
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 == 3:
73
  return csv_filename, "\n".join(progress_info), None
74
+ # return csv_filename, "\n".join(progress_info), None
75
  else:
76
  return "", "No video source selected.", None
77