Update run.py

run.py

@@ -32,17 +32,19 @@ def split_parquet(u, file, batch_size):
     batch_size = int(batch_size)
     if file.lower().endswith(".mp4"):
         print(f"file is: {file}")
-        return file, [file], gr.Slider(value=0, maximum=batch_size-1), gr.Column(visible=True)
+        return file, [[file]], gr.Slider(value=0, maximum=batch_size-1), gr.Column(visible=True)
     file_slice = u.split_parquet(file, batch_size)
     return file_slice[0][0], file_slice, gr.Slider(value=0, maximum=batch_size-1), gr.Column(visible=True)
 
 def select_video(chunks, epoch_idx , batch_idx):
     print(f"chunks: {chunks}, epoch_idx: {epoch_idx}, batch_idx: {batch_idx}")
     chunks_size = len(chunks)
-    if epoch_idx >= chunks_size:
-        return chunks[epoch_idx][batch_idx]
     epoch_idx = int(epoch_idx)
     batch_idx = int(batch_idx)
+    if epoch_idx >= chunks_size:
+        epoch_idx = chunks_size-1
+    if batch_idx >= len(chunks[epoch_idx]):
+        batch_idx = len(chunks[epoch_idx])-1
     return chunks[epoch_idx][batch_idx]
 
 def show_lables():