sync with remote

app.py

@@ -198,18 +198,18 @@ model.summary(print_fn=lambda x: model_summary.write(x + '\n'))
 # Display the model summary in Streamlit
 st.markdown(model_summary.getvalue())
 
-# Save the model plot
-plot_filename = "model_plot.png"
-tf.keras.utils.plot_model(model, to_file=plot_filename, show_shapes=True)
+try:
+    # Save the model plot
+    plot_filename = "model_plot.png"
+    tf.keras.utils.plot_model(model, to_file=plot_filename, show_shapes=True)
+except Exception as e:
+    st.error(f"An error occurred: {e}")
 
 # Streamlit App
 st.title("Model Architecture")
 
-try:
-    # Display the model plot
-    st.image(plot_filename, caption="Neural Network Architecture", use_container_width=True)
-except Exception as e:
-    st.error(f"An error occurred: {e}")
+# Display the model plot
+st.image(plot_filename, caption="Neural Network Architecture", use_container_width=True)
 
 # show a predection, as an example
 show_predictions(test_dataset)