paralym commited on
Commit
2ed44c5
1 Parent(s): ddd6d4d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -216,7 +216,7 @@ def sample_frames_old(video_file, num_frames):
216
  video.release()
217
  return frames
218
 
219
- def sample_frames_frames(video_path, frame_count=32):
220
  video_frames = []
221
  vr = VideoReader(video_path, ctx=cpu(0))
222
  total_frames = len(vr)
@@ -681,7 +681,7 @@ if __name__ == "__main__":
681
  argparser.add_argument("--conv-mode", type=str, default=None)
682
  argparser.add_argument("--temperature", type=float, default=0.7)
683
  argparser.add_argument("--max-new-tokens", type=int, default=4096)
684
- argparser.add_argument("--num_frames", type=int, default=16)
685
  argparser.add_argument("--load-8bit", action="store_true")
686
  argparser.add_argument("--load-4bit", action="store_true")
687
  argparser.add_argument("--debug", action="store_true")
 
216
  video.release()
217
  return frames
218
 
219
+ def sample_frames(video_path, frame_count=32):
220
  video_frames = []
221
  vr = VideoReader(video_path, ctx=cpu(0))
222
  total_frames = len(vr)
 
681
  argparser.add_argument("--conv-mode", type=str, default=None)
682
  argparser.add_argument("--temperature", type=float, default=0.7)
683
  argparser.add_argument("--max-new-tokens", type=int, default=4096)
684
+ argparser.add_argument("--num_frames", type=int, default=32)
685
  argparser.add_argument("--load-8bit", action="store_true")
686
  argparser.add_argument("--load-4bit", action="store_true")
687
  argparser.add_argument("--debug", action="store_true")