Update xgb_mental_health.py

xgb_mental_health.py

@@ -68,7 +68,7 @@ if __name__ == "__main__":
         y_test, y_pred = mental_classifier.train_model(X, y)
         mental_classifier.save_model()
     else:
-        mental_classifier.load_model()
+        mental_classifier.load_model(model_path)
         mental_classifier.initialize_tokenizer(tokenizer_model_name)  # Ensure tokenizer is initialized if loading model from pickle
         mental_classifier.preprocess_data()