lalalalalalalalalala commited on
Commit
13d8c36
·
verified ·
1 Parent(s): ac15feb

Update run.py

Browse files
Files changed (1) hide show
  1. run.py +1 -1
run.py CHANGED
@@ -62,7 +62,7 @@ def fast_caption(sys_prompt, usr_prompt, temp, top_p, max_tokens, model, key, en
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
 
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