Spaces:

Lin-Chen
/

ShareCaptioner-Video

Running on Zero

chenlin commited on Jun 12

Commit

4dcb992

•

1 Parent(s): 49bae8f

fix bug

Files changed (1) hide show

app.py CHANGED Viewed

@@ -272,7 +272,7 @@ def generate_slidingcaptioning(video_path):
 def generate_fastcaptioning(video_path):
  q = 'Here are a few key frames of a video, discribe this video in detail.'
  query = f'[UNUSED_TOKEN_146]user\n{q}[UNUSED_TOKEN_145]\n[UNUSED_TOKEN_146]assistant\n'
- imgs = load_quota_video(video_path, start=start, end=end)
  img = img_process(imgs)
  with torch.cuda.amp.autocast():
  response = model_gen(model, query, img, hd_num=16,

 def generate_fastcaptioning(video_path):
  q = 'Here are a few key frames of a video, discribe this video in detail.'
  query = f'[UNUSED_TOKEN_146]user\n{q}[UNUSED_TOKEN_145]\n[UNUSED_TOKEN_146]assistant\n'
+ imgs = load_quota_video(video_path)
  img = img_process(imgs)
  with torch.cuda.amp.autocast():
  response = model_gen(model, query, img, hd_num=16,