tlord commited on
Commit
5bf3c2a
1 Parent(s): 87af122

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -39,12 +39,12 @@ def titanic(ticket_class, sex, port, fare, age, sibsp, parch):
39
  input_list.append(int(sibsp))
40
  input_list.append(int(parch))
41
  data = {
42
- "pclass": CLASS_TO_VALUE[ticket_class],
43
- "sex": sex,
44
- "embarked": PORT_TO_VALUE[port],
45
- "age": age,
46
- "sibsp": int(sibsp),
47
- "parch": int(parch)
48
  }
49
  df = pd.DataFrame(data)
50
  # 'res' is a list of predictions returned as the label.
 
39
  input_list.append(int(sibsp))
40
  input_list.append(int(parch))
41
  data = {
42
+ "pclass": [CLASS_TO_VALUE[ticket_class]],
43
+ "sex": [sex],
44
+ "embarked": [PORT_TO_VALUE[port]],
45
+ "age": [age],
46
+ "sibsp": [int(sibsp)],
47
+ "parch": [int(parch)],
48
  }
49
  df = pd.DataFrame(data)
50
  # 'res' is a list of predictions returned as the label.