sgpai commited on
Commit
61a22d2
·
verified ·
1 Parent(s): f0b0afb

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -58,5 +58,6 @@ input_data = pd.DataFrame([{
58
 
59
  # Predict button
60
  if st.button("Predict Purchase"):
61
- prediction = 'Yes' if model.predict(input_data)[0] == 1 else 'No'
 
62
  st.write(f"Purchase Prediction Result: {prediction}.")
 
58
 
59
  # Predict button
60
  if st.button("Predict Purchase"):
61
+ #####prediction = 'Yes' if model.predict(input_data)[0] == 1 else 'No'
62
+ prediction = model.predict(input_data)[0]
63
  st.write(f"Purchase Prediction Result: {prediction}.")