Update app.py

app.py

@@ -84,3 +84,9 @@ if st.session_state.autoencoder:
     # Button to display the reconstruction
     if st.button("Display Reconstruction"):
         display_reconstruction(test_index, st.session_state.autoencoder, st.session_state.encoder, st.session_state.x_test)
+
+# Display architecture image
+st.subheader("Autoencoder Architecture")
+architecture_image_path = 'image1.png'  # Path to the uploaded image
+architecture_image = Image.open(architecture_image_path)
+st.image(architecture_image, caption="Autoencoder Architecture")