Update app.py

app.py

@@ -17,11 +17,14 @@ selected_option = st.selectbox('Choose an option:', options)
 conv_layers = [layer for layer in model.layers if isinstance(layer, Conv2D)]
 
 fig = plt.figure(figsize=(12, 4))
+
+layer_ind = options.index(selected_option)
+# selected_layer = conv_layers[layer_ind]
 scaler = MinMaxScaler()
 
 # for i in range(3):
 for j in range(6):
-    layer=conv_layers[0]
+    layer=conv_layers[layer_ind]
     weights=layer.get_weights()[0][:,:,0,j]
     norm_weights = scaler.fit_transform(weights)
     plt.subplot(2,3,j+1)