Update app.py

app.py

@@ -119,18 +119,18 @@ st.write(f"A Collection of ANN Models with a 1-xReLU-1 Architecture for Basic 1D
 #     # Level 6: Neurons count dropdown, based on selected function
 #     neurons = st.selectbox("Select Neurons Count", list(folder_hierarchy[initialisation][sampleSize][batchSize][epochs][functions].keys()))
 
-
-initialisation = st.sidebar.selectbox("Select Initialisation", list(folder_hierarchy.keys()))
-sampleSize = st.sidebar.selectbox("Select Sample Size", list(folder_hierarchy[initialisation].keys()))
-batchSize = st.sidebar.selectbox("Select Batch Size", list(folder_hierarchy[initialisation][sampleSize].keys()))
-epochs = st.sidebar.selectbox("Select Epochs Count", list(folder_hierarchy[initialisation][sampleSize][batchSize].keys()))
-functions = st.sidebar.selectbox("Select Neurons Count", list(folder_hierarchy[initialisation][sampleSize][batchSize][epochs].keys()))
-neurons = st.sidebar.selectbox("Select Neurons Count", list(folder_hierarchy[initialisation][sampleSize][batchSize][epochs][functions].keys()))
+repo = st.sidebar.selectbox("Select Model Repository",list(folder_hierarchy.keys()))
+initialisation = st.sidebar.selectbox("Select Initialisation", list(folder_hierarchy[repo].keys()))
+sampleSize = st.sidebar.selectbox("Select Sample Size", list(folder_hierarchy[repo][initialisation].keys()))
+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 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: {initialisation} : {sampleSize} : {batchSize} : {epochs} : {functions} : {neurons}")
 
-modelPath = os.path.join(os.getcwd(), initialisation, sampleSize, batchSize, epochs, functions, neurons);
+modelPath = os.path.join(os.getcwd(), repo, initialisation, sampleSize, batchSize, epochs, functions, neurons);
 model = keras.models.load_model(modelPath);
 
 visualiser = NNVisualiser(model);