Update main.py

main.py

@@ -28,6 +28,14 @@ app.add_middleware(
 
 def train_the_model(data,page):
     if page==2:
+# Function to evaluate the model
+        def evaluate_model(model, X_test, y_test):
+            y_pred = model.predict(X_test)
+            accuracy = accuracy_score(y_test, y_pred)
+            print("Accuracy:", accuracy)
+            print("Classification Report:\n", classification_report(y_test, y_pred))
+            return accuracy
+            
         new_data = data
         encoders = load('encoders.joblib')
         xgb_model = load('xgb_model.joblib')