Louis VAUBOURDOLLE commited on
Commit
562ddd9
1 Parent(s): 491d788

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -42,7 +42,8 @@ def sentence_builder(credit, age, tenure, balance, nb_prods, has_card, active, e
42
  "EstimatedSalary": est_salary,
43
  }]
44
  df = pd.json_normalize(data)
45
- return bool(pl.predict(df)[0]) ? "User stays in bank" : "User can leave the bank"
 
46
 
47
  iface = gr.Interface(
48
  sentence_builder,
 
42
  "EstimatedSalary": est_salary,
43
  }]
44
  df = pd.json_normalize(data)
45
+ res = bool(pl.predict(df)[0])
46
+ return res == 0 ? 'User will stay' : 'User will exit'
47
 
48
  iface = gr.Interface(
49
  sentence_builder,