Update app.py

app.py

@@ -319,6 +319,8 @@ def generation_loop(video_path='./test_videos', seed=666, fps_out=12, batch_size
                     )[0]
                     / 255.0
                 )
+                if video.size(0) > 121:
+                    video = video[:121]
                 print(f"Read video size: {video.size()}")
                 output_dir = 'output/' + str(uuid.uuid4()) + '.mp4'
             else: