Update app.py

app.py

@@ -47,17 +47,17 @@ y_all = train['Survived']
 num_test = 0.20
 X_train, X_test, y_train, y_test = train_test_split(X_all, y_all, test_size=num_test, random_state=23)
 
-#clf = RandomForestClassifier()
-tree = DecisionTreeClassifier()
-tree.fit(X_train, y_train)
-predictions = tree.predict(X_test)
+clf = RandomForestClassifier()
+#tree = DecisionTreeClassifier()
+clf.fit(X_train, y_train)
+predictions = clf.predict(X_test)
 
 def predict_survival(sex, age, fare):
     df = pd.DataFrame.from_dict({'Sex': [sex], 'Age': [age], 'Fare': [fare]})
     df = encode_sex(df)
     df = encode_fares(df)
     df = encode_ages(df)
-    pred = tree.predict_proba(df)[0]
+    pred = clf.predict_proba(df)[0]
     return {'Perishes': float(pred[0]), 'Survives': float(pred[1])}
     
 sex = gr.inputs.Radio(['female', 'male'], label="Sex")