inie2003 commited on
Commit
bb180db
·
verified ·
1 Parent(s): d4e862b

changed slider for limits

Browse files
Files changed (1) hide show
  1. app.py +6 -8
app.py CHANGED
@@ -13,7 +13,7 @@ AWS_ACCESS_KEY_ID = os.getenv("AWS_ACCESS_KEY_ID")
13
  AWS_SECRET_ACCESS_KEY = os.getenv("AWS_SECRET_ACCESS_KEY")
14
 
15
  # Predefined list of datasets
16
- datasets = ["WayveScenes", "MajorTom-Germany", "MajorTom-Netherlands"]
17
  description = {
18
  "WayveScenes": "A large-scale dataset featuring diverse urban driving scenes, captured from vehicles to advance AI perception and navigation in complex environments.",
19
  "MajorTom-Germany": "A geospatial dataset containing satellite imagery from across Germany, designed for tasks like land-use classification, environmental monitoring, and earth observation analytics.",
@@ -68,13 +68,11 @@ def main():
68
  st.text("Small Object Search Disabled")
69
  st.session_state.dataset_number = st.selectbox("Select Subset of Data", list(range(1, selection[dataset_name][0] + 1)))
70
  st.text(f"You have selected Main Dataset {st.session_state.dataset_number}")
71
- if 'MajorTom' in dataset_name and not st.session_state.search_in_small_objects:
72
- df, total_rows = load_dataset_with_limit(dataset_name, st.session_state.dataset_number, st.session_state.search_in_small_objects, limit=1)
73
- dataset_limit = st.slider("Size of Dataset to be searched from", min_value=1000, max_value=total_rows, value=1500)
74
- st.text(f'The smaller the dataset the faster the search will work.')
75
- else:
76
- dataset_limit = st.slider("Size of Dataset to be searched from", min_value=1000, max_value=70000, value=30000)
77
- st.text(f'The smaller the dataset the faster the search will work.')
78
 
79
  # Load dataset with limit only if not already loaded
80
  if st.button("Load Dataset"):
 
13
  AWS_SECRET_ACCESS_KEY = os.getenv("AWS_SECRET_ACCESS_KEY")
14
 
15
  # Predefined list of datasets
16
+ datasets = ["WayveScenes", "MajorTom-Germany", "MajorTom-Netherlands", "MajorTom-UK"]
17
  description = {
18
  "WayveScenes": "A large-scale dataset featuring diverse urban driving scenes, captured from vehicles to advance AI perception and navigation in complex environments.",
19
  "MajorTom-Germany": "A geospatial dataset containing satellite imagery from across Germany, designed for tasks like land-use classification, environmental monitoring, and earth observation analytics.",
 
68
  st.text("Small Object Search Disabled")
69
  st.session_state.dataset_number = st.selectbox("Select Subset of Data", list(range(1, selection[dataset_name][0] + 1)))
70
  st.text(f"You have selected Main Dataset {st.session_state.dataset_number}")
71
+
72
+ df, total_rows = load_dataset_with_limit(dataset_name, st.session_state.dataset_number, st.session_state.search_in_small_objects, limit=1)
73
+ dataset_limit = st.slider("Size of Dataset to be searched from", min_value=0, max_value=min(total_rows, 80000), value=int(min(total_rows, 80000)/2))
74
+ st.text(f'The smaller the dataset the faster the search will work.')
75
+
 
 
76
 
77
  # Load dataset with limit only if not already loaded
78
  if st.button("Load Dataset"):