Upload folder using huggingface_hub

app.py

@@ -39,17 +39,38 @@ def _np(x):
 
 
 def _read_frames(path, max_frames):
+    """Sample up to `max_frames` frames EVENLY across the whole video (proper intervals), not just the
+    first N. A clip with <= max_frames frames is taken in full; a longer clip is sub-sampled at a
+    constant stride so the selection spans start→end."""
     import imageio
+    max_frames = max(1, int(max_frames))
     reader = imageio.get_reader(path)
-    frames = []
     try:
+        try:
+            n = int(reader.count_frames())
+        except Exception:  # some streams can't report a count -> fall back to sequential
+            n = 0
+        if n > max_frames:
+            # evenly-spaced indices spanning 0 .. n-1 (e.g. 96 frames, max 24 -> every 4th frame)
+            idxs = sorted({round(i * (n - 1) / (max_frames - 1)) for i in range(max_frames)}) \
+                if max_frames > 1 else [0]
+            frames = []
+            for i in idxs:
+                try:
+                    frames.append(np.asarray(reader.get_data(i)))
+                except Exception:
+                    break
+            if frames:
+                return frames
+        # short clip (or no count / seek unsupported): read sequentially up to max_frames
+        frames = []
         for i, fr in enumerate(reader):
             if i >= max_frames:
                 break
             frames.append(np.asarray(fr))
+        return frames
     finally:
         reader.close()
-    return frames
 
 
 @spaces.GPU(duration=300)