Update app.py

app.py

@@ -11,6 +11,8 @@ from sklearn.ensemble import GradientBoostingClassifier
 from sklearn.linear_model import LogisticRegression
 from sklearn.metrics import accuracy_score, precision_score, recall_score, f1_score, roc_auc_score, roc_curve
 import shap
+from xgboost import XGBClassifier
+from lightgbm import LGBMClassifier
 
 def load_data():
     data = pd.read_csv('exported_named_train_good.csv')
@@ -26,7 +28,9 @@ def train_models(X_train, y_train, X_test, y_test):
         "Logistic Regression": LogisticRegression(random_state=42),
         "Decision Tree": DecisionTreeClassifier(random_state=42),
         "Random Forest": RandomForestClassifier(random_state=42),
-        "Gradient Boost": GradientBoostingClassifier(random_state=42)
+        "Gradient Boost": GradientBoostingClassifier(random_state=42),
+        "Extreme Gradient Boosting": XGBClassifier(random_state=42),
+        "Light Gradient Boosting": LGBMClassifier(random_state=42)
     }
     
     results = {}