Upload app.py

app.py

@@ -42,7 +42,7 @@ test.join(pd.DataFrame(model_rf.predict(Y),columns=["predicted"]))[["prognosis",
 st.title('Disease-Prognosis Correlation Visualization')
 
 # Calculate the correlation matrix
-corr = df_train.corr()
+corr = train.corr()
 
 # Create a heatmap to visualize the correlation
 plt.figure(figsize=(10, 8))