pushing batch size
Browse files- app.py +1 -1
- 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=
|
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
|