Upload app.py

app.py

@@ -201,12 +201,17 @@ def generate_prediction(input_data):
     # prediction = MODEL.predict(input_data)
     proba = MODEL.predict_proba(input_data)
     st.write(proba)
-    positive_proba = round(proba[-1]*100)
+    st.write(proba[1])
+    st.write(proba.shape)
+
+    '''
+    positive_proba = round(proba[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