ccclemenfff commited on
Commit
3b9ccb8
·
1 Parent(s): d1059e1
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -112,7 +112,7 @@ class EndpointHandler:
112
 
113
  vr = VideoReader(video_path, ctx=cpu(0))
114
  total_frames = len(vr)
115
- indices = get_index(total_frames, num_segments)
116
  frames = [Image.fromarray(vr[i].asnumpy()) for i in indices]
117
 
118
  transform = T.Compose([
 
112
 
113
  vr = VideoReader(video_path, ctx=cpu(0))
114
  total_frames = len(vr)
115
+ indices = self.get_index(total_frames, num_segments)
116
  frames = [Image.fromarray(vr[i].asnumpy()) for i in indices]
117
 
118
  transform = T.Compose([