Update run.py

run.py

@@ -8,8 +8,8 @@ dataset_options = [
     "OpenVideo/pexels-raw",
     "OpenVideo/Sample-2k",
 ]
-global fetch_files
-global fetch_files_select_idx
+fetch_files = []
+fetch_files_select_idx = 0
 
 def login(token):
     u = HFUser.from_token(token)
@@ -21,8 +21,8 @@ def show_time(u, name):
 def list_dataset(u, repo):
     print(f"repo is: {repo}")
     files = u.list_dataset(repo)
+    global fetch_files
     fetch_files = files
-    fetch_files_select_idx = 0
     return gr.Dropdown(value=files[0], choices=files), gr.Column(visible=True), gr.Column(visible=False)
 
 def fetch_parquet(u, fname):
@@ -56,6 +56,8 @@ def next_chunks(video_chunks, epoch_idx):
     return (epoch_idx+1)%length, gr.Slider(value=0)
 
 def next_mp4_chunks(video_chunks, epoch_idx):
+    global fetch_files_select_idx
+    fetch_files_select_idx = fetch_files_select_idx + 1
     return gr.Dropdown(value=fetch_files[fetch_files_select_idx + 1], choices=fetch_files)
 
 with gr.Blocks(**GR_CONF) as Core: