Upload app.py

app.py

@@ -198,17 +198,20 @@ def generate_prediction(input_data):
 
     -return: predicted probability of having a cardiovascular disease
     """
-    # prediction = MODEL.predict(input_data)
-    proba = MODEL.predict_proba(input_data)    
-    positive_proba = round(proba[0,1]*100)
+    # Compute probability
+    probs = MODEL.predict_proba(input_data)    
+    positive_proba = round(probs[0,1]*100)
 
-    st.title('Predicted cardiovascular disease probability:')
+    # Show results
+    st.markdown("<h2 style='text-align: center; color: black;'>\
+                Predicted cardiovascular disease probability:</h2>",
+                unsafe_allow_html=True)
     st.markdown(f"<h1 style='text-align: center; color: grey;'>\
                 {positive_proba} %</h1>",
                 unsafe_allow_html=True)
     st.progress(positive_proba)
     
-    return proba
+    return probs
 
 
 ###############################
@@ -252,8 +255,7 @@ if __name__ == "__main__":
 
     # Generate prediction
     if IS_CLICKED:
-        st.write('Work in progress!')
-        prediction = generate_prediction(session.input_data)
+        predicted_probs = generate_prediction(session.input_data)
 
     st.text("")
     st.text("")