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=10, n_jobs=-1, scoring='accuracy')
+        grid_search = GridSearchCV(xgb, param_grid, cv=2, n_jobs=-1, scoring='accuracy')
         
         # Fit the grid search to the data
         grid_search.fit(X_train, y_train)
@@ -172,4 +172,5 @@ async def your_continuous_function(page: int,paginate: int,Tenant: str):
 
 @app.get("/test_api")
 async def test_api():
-    return "api_working"
+    encoders = load('encoders.joblib')
+    xgb_model = load('xgb_model.joblib')