Update app.py

app.py

@@ -25,6 +25,7 @@ import requests
 import asyncio
 from io import BytesIO
 import base64
+import seaborn as sns
 import time
 from sklearn.cluster import KMeans
 import scipy.stats as stats
@@ -766,9 +767,9 @@ elif app_mode == "Model Training":
 
                         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)
                         ax_conf.set_xlabel('Predicted Labels')
                         ax_conf.set_ylabel('True Labels')
                         ax_conf.set_title('Confusion Matrix')