Campfireman commited on
Commit
4506881
1 Parent(s): f5d0900

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -26,7 +26,7 @@ def tb_titanic(pclass,sex,age,sibsp,parch,embarked,fare_per_customer,cabin):
26
  input_list.append(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, 1))
30
  return "This guy will"
31
 
32
  demo = gr.Interface(
 
26
  input_list.append(cabin)
27
  # 'res' is a list of predictions returned as the label.
28
  #global res
29
+ res = model.predict(np.asarray(input_list).reshape(8, 1))
30
  return "This guy will"
31
 
32
  demo = gr.Interface(