Update prediction.py

eda.py

@@ -6,7 +6,9 @@ import plotly.express as px
 from PIL import Image
 
 st.set_page_config(
-    page_title="FIFA 2022 - EDA", layout="wide", initial_sidebar_state="expanded"
+    page_title="Heart Failure Classification",
+    layout="wide",
+    initial_sidebar_state="expanded",
 )
 
 # dataset

prediction.py

@@ -65,10 +65,9 @@ def run():
         )
         sc = col1.number_input(
             "Serum Creatinine",
-            min_value=0,
-            max_value=100,
+            min_value=0.0,
+            max_value=100.0,
             value=1.75,
-            step=1e-3,
             help="mg/dL",
         )
         time = col2.number_input(
@@ -100,13 +99,16 @@ def run():
     if submitted:
         y_pred_inf = model_pipeline.predict(data_inf)
         st.write("# Classification Results:")
-        if str(int(y_pred_inf)) == 0:
+
+        if y_pred_inf[0] == 1:
             st.write(
-                "Pasien diprediksi akan meninggal dunia, segera hubungi dokter atau tenaga medis"
+                "### Pasien diprediksi akan meninggal dunia, <span style='color:red;'> segera hubungi dokter atau tenaga medis </span>",
+                unsafe_allow_html=True,
             )
         else:
             st.write(
-                "Pasien **tidak** diprediksi meninggal dunia, untuk konsultasi lebih lanjut, hubungi dokter atau tenaga medis"
+                "### Pasien tidak diprediksi meninggal dunia,<span style='color:green;'> untuk konsultasi lebih lanjut, hubungi dokter atau tenaga medis </span>",
+                unsafe_allow_html=True,
             )