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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -42,7 +42,7 @@ 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])
46
 
47
  iface = gr.Interface(
48
  sentence_builder,
 
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,