Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -216,7 +216,7 @@ def sample_frames_old(video_file, num_frames):
|
|
216 |
video.release()
|
217 |
return frames
|
218 |
|
219 |
-
def
|
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=
|
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")
|