Update media.py

media.py

@@ -103,9 +103,7 @@ def _load_video_with_fps(video_path: str, *, num_frames: int, fps: float) -> Lis
         raise ValueError(f"Video '{video_path}' seems to be empty or corrupted.")
 
     # Compute total frames to sample based on desired fps
-    sampled_frame_count = int(duration_sec * fps)
-    sampled_frame_count = ((sampled_frame_count + 127) // 128) * 128
-    sampled_frame_count = min(sampled_frame_count, num_frames)
+    sampled_frame_count = min(((int(duration_sec * fps) + 127) // 128) * 128, ((num_frames + 127) // 128) * 128)
 
     # Compute which frame indices to sample
     indices = np.linspace(0, frame_count - 1, sampled_frame_count).astype(int)