bugfix: short video index issue

app.py

@@ -103,7 +103,7 @@ with gr.Blocks(theme=gr.themes.Default()) as demo:
             start=lbhd_predictions[i][0], 
             end=lbhd_predictions[i][1],
             output_file=f'{i}_lbhd.mp4'
-        ) for i in range(10)]
+        ) for i in range(min(10, len(lbhd_predictions)))]
         
         return { 
             output_videos: output_files, 

lbhd/infer.py

@@ -20,6 +20,7 @@ scoringModel = checkpoint['model']
 
 
 def sample_clips(x, frames_per_clip: int):
+    print("x shape", x.shape, "frames_per_clip", frames_per_clip)
     x = torch.stack( [ x[i:i+frames_per_clip] for i in range(len(x) - frames_per_clip + 1) ] )
     return x
 
@@ -43,8 +44,8 @@ def sample_frames(file_name, fps):
 def videofile_to_frames(filename, sample_every_second=True):
     clip = VideoFileClip(filename)
     frames = clip.iter_frames()
-    frames = [Image.fromarray(frame) for index, frame in enumerate(frames) if index % clip.fps == 0]
-    print('Number of frames in video is:', len(frames))
+    frames = [Image.fromarray(frame) for index, frame in enumerate(frames) if index % int(clip.fps) == 0]
+    print('clip.fps', clip.fps, 'Number of frames in video is:', len(frames))
     return frames