Spaces:
Runtime error
Runtime error
Update run.py
Browse files
run.py
CHANGED
@@ -57,12 +57,13 @@ def fast_caption(sys_prompt, usr_prompt, temp, top_p, max_tokens, model, key, en
|
|
57 |
_ = f.write(binary)
|
58 |
_chunk.append(_v.name)
|
59 |
|
60 |
-
processor = VideoProcessor(frame_format=frame_format, frame_limit=frame_limit)
|
61 |
-
frames = processor._decode(_v.name)
|
62 |
-
base64_list = processor.to_base64_list(frames)
|
63 |
-
api = AzureAPI(key=key, endpoint=endpoint, model=model, temp=temp, top_p=top_p, max_tokens=max_tokens)
|
64 |
-
caption = api.get_caption(sys_prompt, usr_prompt, base64_list)
|
65 |
writer.writerow({'md5': _v.name, 'caption': caption})
|
|
|
66 |
progress_info.append(f"Processed video with MD5: {md5}")
|
67 |
return csv_filename, "\n".join(progress_info), None
|
68 |
return csv_filename, "\n".join(progress_info), None
|
|
|
57 |
_ = f.write(binary)
|
58 |
_chunk.append(_v.name)
|
59 |
|
60 |
+
# processor = VideoProcessor(frame_format=frame_format, frame_limit=frame_limit)
|
61 |
+
# frames = processor._decode(_v.name)
|
62 |
+
# base64_list = processor.to_base64_list(frames)
|
63 |
+
# api = AzureAPI(key=key, endpoint=endpoint, model=model, temp=temp, top_p=top_p, max_tokens=max_tokens)
|
64 |
+
# caption = api.get_caption(sys_prompt, usr_prompt, base64_list)
|
65 |
writer.writerow({'md5': _v.name, 'caption': caption})
|
66 |
+
writer.writerow({'md5': _v.name, 'caption': 'caption'})
|
67 |
progress_info.append(f"Processed video with MD5: {md5}")
|
68 |
return csv_filename, "\n".join(progress_info), None
|
69 |
return csv_filename, "\n".join(progress_info), None
|