Update app.py

app.py

@@ -760,16 +760,15 @@ elif app_mode == "Model Training":
                         #Confusion Matrix
 
                         conf_matrix = confusion_matrix(y_test, y_pred)
-
-                        #Heatmap
+                        
+                        # Assuming conf_matrix is your confusion matrix
                         fig_conf, ax_conf = plt.subplots()
-                        sns.heatmap(conf_matrix, annot=True, fmt='d', cmap='Blues', ax_conf)
+                        sns.heatmap(conf_matrix, annot=True, fmt='d', cmap='Blues', ax=ax_conf)  # Corrected named argument
                         ax_conf.set_xlabel('Predicted Labels')
                         ax_conf.set_ylabel('True Labels')
                         ax_conf.set_title('Confusion Matrix')
                         st.pyplot(fig_conf)
 
-
                     #Added section for model visualization
                     st.subheader("Model Visualization")
                     #Use conditional to make sure that everything only executes when the data set is trained and not outside of it.