Update pages/6_Logistic_Regression.py

pages/6_Logistic_Regression.py

@@ -50,7 +50,7 @@ For this example, we'll only use the Setosa and Versicolor classes.
 st.write("### Flower Images")
 col1, col2 = st.columns(2)
 with col1:
-    st.image("https://en.wikipedia.org/wiki/File:Irissetosa1.jpg", caption="Iris Setosa", use_column_width=True)
+    st.image("https://upload.wikimedia.org/wikipedia/commons/4/49/Iris_setosa_2.jpg", caption="Iris Setosa", use_column_width=True)
 with col2:
     st.image("https://upload.wikimedia.org/wikipedia/commons/4/41/Iris_versicolor_3.jpg", caption="Iris Versicolor", use_column_width=True)
 
@@ -76,11 +76,14 @@ model.compile(optimizer='adam', loss='binary_crossentropy', metrics=['accuracy']
 # Display the model architecture
 st.write("### Model Architecture")
 st.write(model.summary())
-fig, ax = plt.subplots()
-buf = io.BytesIO()
-plot_model(model, to_file=buf, show_shapes=True, show_layer_names=True)
-buf.seek(0)
-st.image(buf, caption='Logistic Regression Model Architecture', use_column_width=True)
+try:
+    fig, ax = plt.subplots()
+    buf = io.BytesIO()
+    plot_model(model, to_file=buf, show_shapes=True, show_layer_names=True)
+    buf.seek(0)
+    st.image(buf, caption='Logistic Regression Model Architecture', use_column_width=True)
+except ImportError:
+    st.warning('Graphviz is not installed. Model architecture visualization is skipped.')
 
 # Train the model
 model.fit(X_train, y_train, epochs=epochs, verbose=0)