Update app.py

app.py

@@ -62,9 +62,13 @@ def main():
         st.text("Small Object Search Disabled")
         st.session_state.dataset_number = st.selectbox("Select Subset of Data", list(range(1, selection[dataset_name][0] + 1)))
         st.text(f"You have selected Main Dataset {st.session_state.dataset_number}")
-
-    dataset_limit = st.slider("Size of Dataset to be searched from", min_value=1000, max_value=30000, value=10000)
-    st.text(f'The smaller the dataset the faster the search will work.')
+    if 'MajorTom' in dataset_name:
+        df, total_rows = load_dataset_with_limit(dataset_name, st.session_state.dataset_number, st.session_state.search_in_small_objects, limit=1)
+        dataset_limit = st.slider("Size of Dataset to be searched from", min_value=1000, max_value=total_rows, value=1500)
+        st.text(f'The smaller the dataset the faster the search will work.')
+    else:
+        dataset_limit = st.slider("Size of Dataset to be searched from", min_value=1000, max_value=30000, value=10000)
+        st.text(f'The smaller the dataset the faster the search will work.')
 
     # Load dataset with limit only if not already loaded
     if st.button("Load Dataset"):