Update main.py

main.py

@@ -94,7 +94,7 @@ def train_the_model(data):
         xgb = XGBClassifier(use_label_encoder=False, eval_metric='logloss')
         
         # Setup GridSearchCV
-        grid_search = GridSearchCV(xgb, param_grid, cv=2, n_jobs=-1, scoring='accuracy')
+        grid_search = GridSearchCV(xgb, param_grid, cv=40, n_jobs=-1, scoring='accuracy')
         
         # Fit the grid search to the data
         grid_search.fit(X_train, y_train)
@@ -196,8 +196,8 @@ def predict(
 
     try:
         # Load your trained model and encoders
-        xgb_model = load('transexpress_xgb_model_v11.joblib')
-        encoders = load('transexpress_encoders_v11.joblib')
+        xgb_model = load('transexpress_xgb_model.joblib')
+        encoders = load('transexpress_encoders.joblib')
     except:
         return {"no model found so first trained the model using data fecther"}