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')