Update app.py

app.py

@@ -85,8 +85,8 @@ if st.session_state.autoencoder:
     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)
+architecture_image = cv2.imread("image1.png")
+
+architecture_image = cv2.cvtColor(architecture_image, cv2.COLOR_BGR2RGB)  # Convert to RGB for proper display
+
 st.image(architecture_image, caption="Autoencoder Architecture")