Spaces:

gaspar-avit
/

Cardiovascular-Disease-prediction

Runtime error

gaspar-avit commited on Apr 10, 2023

Commit

1b3d335

1 Parent(s): d527598

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[1])
     st.write(proba.shape)
     '''