mfarre HF staff commited on
Commit
3148909
·
1 Parent(s): 7c6ea46

pushing batch size

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. video_highlight_detector.py +1 -1
app.py CHANGED
@@ -79,7 +79,7 @@ def process_video(
79
 
80
  progress(0.1, desc="Loading model...")
81
  model, processor = load_model()
82
- detector = BatchedVideoHighlightDetector(model, processor)
83
 
84
  progress(0.2, desc="Analyzing video content...")
85
  video_description = detector.analyze_video_content(video_path)
 
79
 
80
  progress(0.1, desc="Loading model...")
81
  model, processor = load_model()
82
+ detector = BatchedVideoHighlightDetector(model, processor, batch_size=32)
83
 
84
  progress(0.2, desc="Analyzing video content...")
85
  video_description = detector.analyze_video_content(video_path)
video_highlight_detector.py CHANGED
@@ -549,7 +549,7 @@ class BatchedVideoHighlightDetector:
549
  start_sec=0,
550
  end_sec=duration,
551
  target_fps=sample_rate,
552
- max_frames=32 # Limit total frames to not overwhelm model
553
  )
554
 
555
  # Build conversation asking for video description
 
549
  start_sec=0,
550
  end_sec=duration,
551
  target_fps=sample_rate,
552
+ max_frames=64
553
  )
554
 
555
  # Build conversation asking for video description