FrankAst commited on
Commit
5230a4e
1 Parent(s): b9c4491

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -51,7 +51,7 @@ clf = RandomForestClassifier()
51
  clf.fit(X_train, y_train)
52
  predictions = clf.predict(X_test)
53
 
54
- def predict_survival(sex, age, fare):
55
  df = pd.DataFrame.from_dict({'Sex': [sex], 'Age': [age], 'Fare': [fare]})
56
  df = encode_sex(df)
57
  df = encode_fares(df)
@@ -64,5 +64,5 @@ age = gr.inputs.Slider(minimum=0, maximum=120, default=22, label="Age")
64
  fare = gr.inputs.Slider(minimum=0, maximum=200, default=100, label="Fare (british pounds)")
65
  height = gr.inputs.Slider(minimum=0.1, maximum = 2.10, default=1.75, label="Height")
66
 
67
- gr.Interface(predict_survival, [sex, age, fare], "label", live=True, thumbnail="https://raw.githubusercontent.com/gradio-app/hub-titanic/master/thumbnail.png", analytics_enabled=False,
68
  title="Surviving the Titanic", description="Demographics importance on Titanic Survival probabilities:").launch();
 
51
  clf.fit(X_train, y_train)
52
  predictions = clf.predict(X_test)
53
 
54
+ def predict_survival(sex, age, fare,height):
55
  df = pd.DataFrame.from_dict({'Sex': [sex], 'Age': [age], 'Fare': [fare]})
56
  df = encode_sex(df)
57
  df = encode_fares(df)
 
64
  fare = gr.inputs.Slider(minimum=0, maximum=200, default=100, label="Fare (british pounds)")
65
  height = gr.inputs.Slider(minimum=0.1, maximum = 2.10, default=1.75, label="Height")
66
 
67
+ gr.Interface(predict_survival, [sex, age, fare,height], "label", live=True, thumbnail="https://raw.githubusercontent.com/gradio-app/hub-titanic/master/thumbnail.png", analytics_enabled=False,
68
  title="Surviving the Titanic", description="Demographics importance on Titanic Survival probabilities:").launch();