lalalalalalalalalala commited on
Commit
6bba632
·
verified ·
1 Parent(s): 5ab6a59

Update run.py

Browse files
Files changed (1) hide show
  1. run.py +7 -10
run.py CHANGED
@@ -49,17 +49,14 @@ def fast_caption(sys_prompt, usr_prompt, temp, top_p, max_tokens, model, key, en
49
  video = df['video'][0]
50
 
51
  md5 = hashlib.md5(video).hexdigest()
52
- with tempfile.NamedTemporaryFile(mode='w+t', delete=True) as temp_video_file:
53
- temp_video_file.write(video)
54
- video_path = temp_video_file.name
55
- print(video_path)
56
 
57
- processor = VideoProcessor(frame_format=frame_format, frame_limit=frame_limit)
58
- frames = processor._decode(video_path)
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)
62
- all_captions.append(caption)
63
 
64
  return "\n\n\n".join(all_captions), f"Processed {len(video_paths)} videos.", None
65
 
 
49
  video = df['video'][0]
50
 
51
  md5 = hashlib.md5(video).hexdigest()
52
+ print(md5)
 
 
 
53
 
54
+ processor = VideoProcessor(frame_format=frame_format, frame_limit=frame_limit)
55
+ frames = processor._decode(video_path)
56
+ base64_list = processor.to_base64_list(frames)
57
+ api = AzureAPI(key=key, endpoint=endpoint, model=model, temp=temp, top_p=top_p, max_tokens=max_tokens)
58
+ caption = api.get_caption(sys_prompt, usr_prompt, base64_list)
59
+ all_captions.append(caption)
60
 
61
  return "\n\n\n".join(all_captions), f"Processed {len(video_paths)} videos.", None
62