Spaces:

gaspar-avit
/

Cardiovascular-Disease-prediction

Runtime error

gaspar-avit commited on Apr 10, 2023

Commit

60319f9

1 Parent(s): 1b3d335

Upload app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -201,7 +201,7 @@ def generate_prediction(input_data):
     # prediction = MODEL.predict(input_data)
     proba = MODEL.predict_proba(input_data)
     st.write(proba)
-    #st.write(proba[1])
     st.write(proba.shape)
     '''

     # prediction = MODEL.predict(input_data)
     proba = MODEL.predict_proba(input_data)
     st.write(proba)
+    st.write(proba[0,1])
     st.write(proba.shape)
     '''