Update main.py

main.py

@@ -8,7 +8,7 @@ import httpx
 import pandas as pd
 from sklearn.model_selection import train_test_split, GridSearchCV
 from sklearn.preprocessing import LabelEncoder
-from xgboost import XGBClassifierS
+from xgboost import XGBClassifier
 from sklearn.metrics import accuracy_score, classification_report
 
 app = FastAPI()
@@ -35,11 +35,6 @@ app.add_middleware(
 from joblib import dump
 
 def train_the_model(data):
-    import pandas as pd
-    from sklearn.model_selection import train_test_split, GridSearchCV
-    from sklearn.preprocessing import LabelEncoder
-    from xgboost import XGBClassifierS
-    from sklearn.metrics import accuracy_score, classification_report
     data = data
     
     # Select columns