Spaces:

OrbitalLab
/

SportsAI

Paused

nicolasbuitragob commited on Apr 13

Commit

c903b19

1 Parent(s): 1988c5a

remove frame limit

Files changed (1) hide show

vitpose.py CHANGED Viewed

@@ -38,7 +38,7 @@ class VitPose:
             if not ret:
                 break
             frames.append(frame)
-        return frames[:2]
     def run(self,video):
         frames = self.video_to_frames(video)

             if not ret:
                 break
             frames.append(frame)
+        return frames
     def run(self,video):
         frames = self.video_to_frames(video)