Update app.py

app.py

@@ -195,21 +195,6 @@ user_input = np.array([age, bp, sg, al, sugar, rbc, pc, pcc, bac, bgr, bu, sc,
 # import dataset
 def get_dataset():
     data = pd.read_csv('kidney.csv')
-
-    # Calculate the correlation matrix
-    # corr_matrix = data.corr()
-
-    # Create a heatmap of the correlation matrix
-    plt.figure(figsize=(10, 8))
-    sns.heatmap(corr_matrix, annot=True, cmap='coolwarm')
-    plt.title('Correlation Matrix')
-    plt.xticks(rotation=45)
-    plt.yticks(rotation=0)
-    plt.tight_layout()
-
-    # Display the heatmap in Streamlit
-     st.pyplot()
-
     return data
 
 def generate_model_labels(model_names):