Upload app.py

app.py

@@ -199,19 +199,15 @@ def generate_prediction(input_data):
     -return: predicted probability of having a cardiovascular disease
     """
     # prediction = MODEL.predict(input_data)
-    proba = MODEL.predict_proba(input_data)
-    st.write(proba)
-    st.write(proba[0,1])
-    st.write(proba.shape)
-
-    '''
-    positive_proba = round(proba[1]*100)
+    proba = MODEL.predict_proba(input_data)    
+    positive_proba = round(proba[0,1]*100)
+    
     st.markdown(f"<h1 style='text-align: center; color: grey;'>\
                 {positive_proba}</h1>",
                 unsafe_allow_html=True)
     st.progress(positive_proba,
                 text='predicted cardiovascular disease probability.')
-    '''
+    
     return proba