Update app.py

app.py

@@ -6,6 +6,7 @@ from sklearn import preprocessing
 from sklearn.model_selection import train_test_split
 from sklearn.ensemble import RandomForestClassifier
 from sklearn.neural_network import MLPClassifier
+from sklearn.tree import  DecisionTreeClassifier
 from sklearn.metrics import accuracy_score
 
 data = pd.read_csv('https://raw.githubusercontent.com/gradio-app/titanic/master/train.csv')
@@ -47,9 +48,9 @@ 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()
-MLP = MLPClassifier()
-MLP.fit(X_train, y_train)
-predictions = MLP.predict(X_test)
+tree = DecisionTreeClassifier()
+tree.fit(X_train, y_train)
+predictions = tree.predict(X_test)
 
 def predict_survival(sex, age, fare):
     df = pd.DataFrame.from_dict({'Sex': [sex], 'Age': [age], 'Fare': [fare]})