Update app.py

app.py

@@ -762,7 +762,7 @@ elif app_mode == "Model Training":
 
                         #Heatmap
                         fig_conf, ax_conf = plt.subplots()
-                        sns.heatmap(conf_matrix, annot=True, fmt='d', cmap='Blues', ax_conf)
+                        sns.heatmap(conf_matrix, ax=ax_conf, annot=True, fmt='d', cmap='Blues')
                         ax_conf.set_xlabel('Predicted Labels')
                         ax_conf.set_ylabel('True Labels')
                         ax_conf.set_title('Confusion Matrix')
@@ -850,6 +850,7 @@ elif app_mode == "Model Training":
                      st.write(f"Accuracy: {accuracy:.4f}")
             except Exception as e: #local error
                  st.error(f"An error occurred during model evaluation: {e}")
+                
 elif app_mode == "Predictions":
     st.title("🔮 Make Predictions")