Update app.py

app.py

@@ -50,7 +50,7 @@ def downloadKerasModel():
 
 # Function to build folder hierarchy up to the 6th level (excluding files and hidden folders)
 # @st.cache_data
-def generate_folder_hierarchy(root_folder, max_depth=6):
+def generate_folder_hierarchy(root_folder, max_depth=7):
     folder_dict = {}
 
     # Traverse through the directory tree
@@ -125,7 +125,7 @@ sampleSize = st.sidebar.selectbox("Select Sample Size", list(folder_hierarchy[re
 batchSize = st.sidebar.selectbox("Select Batch Size", list(folder_hierarchy[repo][initialisation][sampleSize].keys()))
 epochs = st.sidebar.selectbox("Select Epochs Count", list(folder_hierarchy[repo][initialisation][sampleSize][batchSize].keys()))
 functions = st.sidebar.selectbox("Select Function", list(folder_hierarchy[repo][initialisation][sampleSize][batchSize][epochs].keys()))
-neurons = st.sidebar.selectbox("Select Neurons Count", list(folder_hierarchy[repo][initialisation][sampleSize][batchSize][epochs].keys()))
+neurons = st.sidebar.selectbox("Select Neurons Count", list(folder_hierarchy[repo][initialisation][sampleSize][batchSize][epochs][functions].keys()))
 
 # Display the selected values
 st.write(f"You selected: {repo} : {initialisation} : {sampleSize} : {batchSize} : {epochs} : {functions} : {neurons}")