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)
|