lalalalalalalalalala commited on
Commit
b0e0df4
·
verified ·
1 Parent(s): 218bfc6

Update run.py

Browse files
Files changed (1) hide show
  1. run.py +3 -1
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(video)
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)