Update app.py
Browse files
app.py
CHANGED
@@ -9,7 +9,8 @@ classification_model_id = "MCG-NJU/videomae-base"
|
|
9 |
|
10 |
# Object detection model (you can replace this with a more accurate one if needed)
|
11 |
object_detection_model = "yolov5s"
|
12 |
-
|
|
|
13 |
def analyze_video(video):
|
14 |
# Extract key frames from the video using OpenCV
|
15 |
frames = extract_key_frames(video)
|
|
|
9 |
|
10 |
# Object detection model (you can replace this with a more accurate one if needed)
|
11 |
object_detection_model = "yolov5s"
|
12 |
+
TARGET_FRAME_COUNT = 16
|
13 |
+
FRAME_SIZE = (224, 224)
|
14 |
def analyze_video(video):
|
15 |
# Extract key frames from the video using OpenCV
|
16 |
frames = extract_key_frames(video)
|