Update run.py

run.py

@@ -60,13 +60,13 @@ def fast_caption(sys_prompt, usr_prompt, temp, top_p, max_tokens, model, key, en
                             _chunk.append(_v.name)
                             md5 = hashlib.md5(binary).hexdigest()
                             
-                            # processor = VideoProcessor(frame_format=frame_format, frame_limit=frame_limit)
-                            # frames = processor._decode(_v.name)
-                            # base64_list = processor.to_base64_list(frames)
-                            # api = AzureAPI(key=key, endpoint=endpoint, model=model, temp=temp, top_p=top_p, max_tokens=max_tokens)
-                            # caption = api.get_caption(sys_prompt, usr_prompt, base64_list)
-                            #writer.writerow({'md5': _v.name, 'caption': caption})
-                            writer.writerow({'md5': md5, 'caption': 'caption'})
+                            processor = VideoProcessor(frame_format=frame_format, frame_limit=frame_limit)
+                            frames = processor._decode(_v.name)
+                            base64_list = processor.to_base64_list(frames)
+                            api = AzureAPI(key=key, endpoint=endpoint, model=model, temp=temp, top_p=top_p, max_tokens=max_tokens)
+                            caption = api.get_caption(sys_prompt, usr_prompt, base64_list)
+                            writer.writerow({'md5': md5, 'caption': caption})
+                            # writer.writerow({'md5': md5, 'caption': 'caption'})
                             progress_info.append(f"Processed video with MD5: {md5}")
                 return csv_filename, "\n".join(progress_info), None
             else: