Campfireman commited on
Commit
c6dfe1a
1 Parent(s): ed6ea16

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -27,7 +27,7 @@ def tb_titanic(pclass,sex,age,sibsp,parch,embarked,fare_per_customer,cabin):
27
  # 'res' is a list of predictions returned as the label.
28
  #global res
29
  res = model.predict(np.asarray(input_list).reshape(1, 8))
30
- return ("This guy will"+(" survive. " if res[0]==1 else " die. "))
31
 
32
  demo = gr.Interface(
33
  fn=tb_titanic,
 
27
  # 'res' is a list of predictions returned as the label.
28
  #global res
29
  res = model.predict(np.asarray(input_list).reshape(1, 8))
30
+ return ("This guy will"+(" survive. " if res[0]=="S" else " die. "))
31
 
32
  demo = gr.Interface(
33
  fn=tb_titanic,