agayabag commited on
Commit
e6c0164
1 Parent(s): 8d97f5b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -84,7 +84,7 @@ data_final = preprocess.transform(data)
84
 
85
  if st.button('Predict'):
86
  prediction = model.predict(data_final).tolist()[0]
87
- prediction = np.where(prediction >= 0.5, 1, 0)
88
 
89
  if prediction == 0:
90
  prediction = 'Congratulations, this person most likely will stay!'
 
84
 
85
  if st.button('Predict'):
86
  prediction = model.predict(data_final).tolist()[0]
87
+ # prediction = np.where(prediction >= 0.5, 1, 0)
88
 
89
  if prediction == 0:
90
  prediction = 'Congratulations, this person most likely will stay!'