lalalalalalalalalala commited on
Commit
ac15feb
·
verified ·
1 Parent(s): 12c9224

Update run.py

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