Update main.py

main.py

@@ -35,6 +35,11 @@ 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