commit

app.py

@@ -61,7 +61,7 @@ with demo:
         slider = gr.Slider(minimum=0, maximum=dataset_size)
         with gr.Row():
             index = slider.value
-            if index > dataset_size:
+            if index >= dataset_size:
                 index = 0
             target_size = wmtis[index]['normal_image'].size
             natural_outputs = add_column_by_key('natural', target_size)