Update prediction.py

prediction.py

@@ -43,18 +43,18 @@ def run():
         submitted = st.form_submit_button('Prediction')
 
     data_inf = {
-        'age': 42.0,
-        'anaemia': 1,
-        'creatinine_phosphokinase': 582,
-        'diabetes': 0,
-        'ejection_fraction': 25,
-        'high_blood_pressure': 1,
-        'platelets': 302000.0,
-        'serum_creatinine': 0.9,
-        'serum_sodium': 145,
-        'sex': 1,
-        'smoking': 1,
-        'time': 215
+        'age': Age,
+        'anaemia': Anaemia,
+        'creatinine_phosphokinase': Creatinine,
+        'diabetes': Diabetes,
+        'ejection_fraction': Ejection,
+        'high_blood_pressure': BloodPleasure,
+        'platelets': Platelets,
+        'serum_creatinine': SerumCreatinine,
+        'serum_sodium': Serum_Sodium,
+        'sex': Gender,
+        'smoking': Smoking,
+        'time': Time
     }
 
     data_inf = pd.DataFrame([data_inf])