Update app.py

app.py

@@ -28,7 +28,7 @@ def load_page(split):
     st.write(f"# {split}")
     st.caption(f"Contributors: {','.join(contribution_dict[split])}")
     with st.form("dataset_form"):
-        index = st.slider('Select a row', 0, len(ds['train'])-1, 0)
+        index = st.slider('Select a row', 0, len(ds)-1, 0)
         if st.form_submit_button("Load"):
             st.write(f"Row {index}")
             data = ds[index]