Update run.py

run.py

@@ -70,7 +70,7 @@ def fast_caption(sys_prompt, usr_prompt, temp, top_p, max_tokens, model, key, en
                             # writer.writerow({'md5': md5, 'caption': 'caption'})
                             progress_info.append(f"Processed video with MD5: {md5}")
                             i += 1
-                        if i == 2:
+                        if i == 1:
                             break
                         #return csv_filename, "\n".join(progress_info), None
                 return csv_filename, "\n".join(progress_info), None