Update main.py

main.py

@@ -64,7 +64,7 @@ def train_the_model(data):
         }
         
         # Initializing GridSearchCV
-        grid_search = GridSearchCV(estimator=xgb_model, param_grid, cv=50, n_jobs=-1, scoring='accuracy')
+        grid_search = GridSearchCV(xgb_model, param_grid, cv=50, n_jobs=-1, scoring='accuracy')
         
         # Fitting GridSearchCV
         grid_search.fit(X_train, y_train)