MNGames commited on
Commit
871946a
1 Parent(s): 29cb7aa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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)