Update app.py

app.py

@@ -85,7 +85,7 @@ def load_model_and_data():
     X = data.drop("Target", axis=1)
     y = data['Target']
 
-    return model, X, y
+    return model, X, y, X.columns
     
 import streamlit as st
 import pandas as pd