diff --git a/ML/model_train.py b/ML/model_train.py
index 4a578b5..677661b 100644
--- a/ML/model_train.py
+++ b/ML/model_train.py
@@ -115,7 +115,7 @@ def create_dataset():
     home_loan_val_final.to_csv('home_loan_val.csv', index=False)
 
 def main() -> None:
-    create_dataset()
+    # create_dataset()
     parser = argparse.ArgumentParser(description='A test program.')
     parser.add_argument("--epochs", help="Prints the supplied argument.", default='10')
     args = parser.parse_args()