fix slider

app.py

@@ -1,5 +1,4 @@
 import math
-import random
 import gradio as gr
 import os
 import json
@@ -71,18 +70,21 @@ def get_col(example, dataset_name):
     return inputs_left, text_inputs_right
 
 with demo:
-    gr.Markdown("# Slide to iterate videos")
-
     with gr.Column():
         dataset_dropdown = gr.Dropdown(choices=list(json_files.keys()), label="Select Dataset", value="Kinetics700")
-        slider = gr.Slider(minimum=0, maximum=math.ceil(500 / batch_size), step=1, label='Page')  # Assuming 500 samples per dataset
+        
+        # Load the selected dataset to determine the number of samples
+        dataset = dataset_dropdown.value
+        with open(json_files[dataset], 'r') as f:
+            data = json.load(f)
+        num_samples = len(data)
+        
+        slider = gr.Slider(minimum=0, maximum=math.floor(num_samples / batch_size) - 1, step=1, label='Page')
+
         with gr.Row():
             index = slider.value
-            dataset = dataset_dropdown.value
             start_index = 0 * batch_size
             end_index = start_index + batch_size
-            with open(json_files[dataset], 'r') as f:
-                data = json.load(f)
             all_examples = data[start_index:end_index]
             all_inputs_left_right = []
             for example_idx, example in enumerate(all_examples):
@@ -93,4 +95,4 @@ with demo:
     slider.change(func, inputs=[slider, dataset_dropdown], outputs=all_inputs_left_right)
     dataset_dropdown.change(func, inputs=[slider, dataset_dropdown], outputs=all_inputs_left_right)
 
-demo.launch()
+demo.launch()