Update run.py

run.py

@@ -69,10 +69,10 @@ 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 == 1:
-                        #     break
-                        return csv_filename, "\n".join(progress_info), None
-                # return csv_filename, "\n".join(progress_info), None
+                        if i == 1:
+                            break
+                        # return csv_filename, "\n".join(progress_info), None
+                return csv_filename, "\n".join(progress_info), None
             else:
                 return "", "No video source selected.", None