Update app.py

app.py

@@ -162,7 +162,7 @@ if uploaded_img is not None:
     st.image(img_resized, caption="Uploaded Image (Resized to 28x28)", use_container_width=True, channels="GRAY")
 
     # Create intermediate model to get output from selected conv layer
-    func_model = Model(inputs=model.input, outputs=selected_layer.output)
+    func_model = Model(inputs=model.layers[0].input, outputs=selected_layer.output)
     fm = func_model.predict(input_img)[0]  # shape: (H, W, C)
 
     # Display feature maps