Update app.py

app.py

@@ -29,18 +29,19 @@ def main():
         # clicked==True only when the button is clicked
         clicked = st.form_submit_button("Predict Stroke")
         if clicked:
-            result=model.predict(pd.DataFrame({"Gender": [gender],
-                                               "Age": [age],
-                                               "Hypertension": [hypertension],
-                                               "Heart_Disease": [heart_disease],
-                                               "Work_type": [work_type],
-                                               "Residence": [residence],
-                                               "Avg_Glucose_Level": [glucose_level],
-                                               "BMI": [bmi],
-                                               "Smoking": [smoking]}))
+            result=model.predict(pd.DataFrame({"gender": [gender],
+                                               "age": [age],
+                                               "hypertension": [hypertension],
+                                               "heart_disease": [heart_disease],
+                                               "work_type": [work_type],
+                                               "Resident_type": [residence],
+                                               "avg_glucose_level": [glucose_level],
+                                               "bmi": [bmi],
+                                               "smoking_status": [smoking]}))
             # Show prediction
             result = 'Yes' if result[0] == 1 else 'No'
             st.success("Your predicted Stroke is "+result)
+            
 # Run main()
 if __name__ == "__main__":
     main()