Update app.py

app.py

@@ -17,6 +17,7 @@ sex = st.selectbox("Sex", ["Male", "Female"])
 chest_pain = st.selectbox("Chest Pain", ["Typical Angina", "Atypical Angina", "Non-Anginal pain", "Asymptomatic"])
 Max_heartrate_achieved = st.number_input("Maximum Heartrate Achieved ")
 Exercice_angina = st.selectbox("Exercise Induced Angina", ["Yes", "No"])
+oldpeak = st.number_input("oldpeak")
 slop = st.selectbox("Slope of the peak exercise ST segment", ["Unsloping", "flat", "Downsloping"])
 nb_vessels = st.selectbox("Number of major vessels", ["0", "1", "2", "3"])
 thal = st.selectbox("Thalassemia", ["Null", "Fixed defect", "Normal", "Reversible defect"])
@@ -31,7 +32,7 @@ if st.button("Predict"):
     nb_vessels = 0 if nb_vessels == "0" else 1 if nb_vessels == "1" else 2 if nb_vessels == "2" else 3
     thal = 0 if thal == "Null" else 1 if thal == "Fixed defect" else 2 if thal == "Normal" else 3 if thal == "Reversible defect" else 0
 
-    input_data = np.array([[sex, chest_pain, Max_heartrate_achieved, Exercice_angina, slop, nb_vessels, thal]])
+    input_data = np.array([[sex, chest_pain, Max_heartrate_achieved, Exercice_angina,oldpeak, slop, nb_vessels, thal]])
 
     # Make a prediction on the user input
     prediction = loaded_xgb_model.predict(input_data)