hysts HF Staff commited on
Commit
899c724
·
1 Parent(s): b350359
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -20,6 +20,8 @@ TEXT_VIDEO_MODEL = Sam3VideoModel.from_pretrained(MODEL_ID).to(DEVICE, dtype=DTY
20
  TEXT_VIDEO_PROCESSOR = Sam3VideoProcessor.from_pretrained(MODEL_ID)
21
  print("Models loaded successfully!")
22
 
 
 
23
 
24
  def try_load_video_frames(video_path_or_url: str) -> tuple[list[Image.Image], dict]:
25
  cap = cv2.VideoCapture(video_path_or_url)
@@ -154,7 +156,6 @@ def init_video_session(
154
  if len(frames) == 0:
155
  raise gr.Error("No frames could be loaded from the video.")
156
 
157
- MAX_SECONDS = 8.0
158
  trimmed_note = ""
159
  fps_in = info.get("fps")
160
  max_frames_allowed = int(MAX_SECONDS * fps_in) if fps_in else len(frames)
 
20
  TEXT_VIDEO_PROCESSOR = Sam3VideoProcessor.from_pretrained(MODEL_ID)
21
  print("Models loaded successfully!")
22
 
23
+ MAX_SECONDS = 8.0
24
+
25
 
26
  def try_load_video_frames(video_path_or_url: str) -> tuple[list[Image.Image], dict]:
27
  cap = cv2.VideoCapture(video_path_or_url)
 
156
  if len(frames) == 0:
157
  raise gr.Error("No frames could be loaded from the video.")
158
 
 
159
  trimmed_note = ""
160
  fps_in = info.get("fps")
161
  max_frames_allowed = int(MAX_SECONDS * fps_in) if fps_in else len(frames)