Update app.py

app.py

@@ -17,7 +17,7 @@ training_data["BPlabel_encoded"] = training_data["BPlabel"].map(
     {"most likely": 1, "probable": 0.75, "least likely": 0.1}
 )
 Y = training_data["BPlabel_encoded"]
-X = training_data.drop(["BPlabel_encoded","BPlabel"], axis=1)
+X = training_data.drop(columns=["BPlabel_encoded","BPlabel"])
 xgb = xgboost.XGBRegressor(
     n_estimators=40,
     learning_rate=0.2,