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

Update run.py

Browse files
Files changed (1) hide show
  1. run.py +1 -1
run.py CHANGED
@@ -53,7 +53,7 @@ def fast_caption(sys_prompt, usr_prompt, temp, top_p, max_tokens, model, key, en
53
  print(md5)
54
 
55
  processor = VideoProcessor(frame_format=frame_format, frame_limit=frame_limit)
56
- frames = processor._decode(video_path)
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)
 
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)