Spaces:
Sleeping
Sleeping
Update run.py
Browse files
run.py
CHANGED
@@ -51,9 +51,11 @@ def fast_caption(sys_prompt, usr_prompt, temp, top_p, max_tokens, model, key, en
|
|
51 |
|
52 |
md5 = hashlib.md5(video).hexdigest()
|
53 |
print(md5)
|
|
|
|
|
54 |
|
55 |
processor = VideoProcessor(frame_format=frame_format, frame_limit=frame_limit)
|
56 |
-
frames = processor._decode(
|
57 |
base64_list = processor.to_base64_list(frames)
|
58 |
api = AzureAPI(key=key, endpoint=endpoint, model=model, temp=temp, top_p=top_p, max_tokens=max_tokens)
|
59 |
caption = api.get_caption(sys_prompt, usr_prompt, base64_list)
|
|
|
51 |
|
52 |
md5 = hashlib.md5(video).hexdigest()
|
53 |
print(md5)
|
54 |
+
with open(md5, 'wb') as f:
|
55 |
+
fc = f.write(video)
|
56 |
|
57 |
processor = VideoProcessor(frame_format=frame_format, frame_limit=frame_limit)
|
58 |
+
frames = processor._decode(md5)
|
59 |
base64_list = processor.to_base64_list(frames)
|
60 |
api = AzureAPI(key=key, endpoint=endpoint, model=model, temp=temp, top_p=top_p, max_tokens=max_tokens)
|
61 |
caption = api.get_caption(sys_prompt, usr_prompt, base64_list)
|