EAV123 commited on
Commit
e6d34bb
·
verified ·
1 Parent(s): 00340e8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -122,7 +122,7 @@ def main():
122
  st.write(user_input)
123
 
124
  prediction = model.predict(user_input)
125
- prediction_proba = model.predict_proba(user_input)[:, 1]
126
 
127
  st.subheader('Prediction:')
128
  if prediction[0] == 1:
 
122
  st.write(user_input)
123
 
124
  prediction = model.predict(user_input)
125
+ prediction_proba = model.predict(user_input, raw_score=True)
126
 
127
  st.subheader('Prediction:')
128
  if prediction[0] == 1: