Update run.py

run.py

@@ -1,6 +1,7 @@
 import gradio as gr
 from huggingface.user import HFUser, GR_CONF
 
+
 Theme = gr.Theme.load(GR_CONF["theme"])
 GR_CONF["theme"] = Theme
 dataset_options = [
@@ -36,6 +37,10 @@ def split_parquet(u, 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)
     return chunks[epoch_idx][batch_idx]
@@ -70,7 +75,7 @@ with gr.Blocks(**GR_CONF) as Core:
                 _list = gr.Button("List", variant='secondary', size='sm')
 
             with gr.Column(visible=False) as Batch:
-                file = gr.Dropdown(label="Parquet")
+                file = gr.Dropdown(label="Parquet/MP4")
                 with gr.Row():
                     _cache= gr.Textbox("Downloading", label="Cache")
                     batch_size= gr.Textbox("8", label="Batch")
@@ -86,7 +91,7 @@ with gr.Blocks(**GR_CONF) as Core:
             submit = gr.Button("Submit", variant="primary", size="sm", scale=1)
             with gr.Row():
                 rst = gr.Button("Reset", variant="stop", size="sm", scale=1)
-                nxt = gr.Button("Next Batch", variant="secondary", size="sm", scale=1)
+                nxt = gr.Button("Next Batch/MP4", variant="secondary", size="sm", scale=1)
 
     _auth.click(fn=login, inputs=_token, outputs=[user, name,  Auth])
     name.change(fn=show_time, inputs=[user, name], outputs=[time, Repo])