update __main__

script.py

@@ -99,4 +99,4 @@ def train_model(training_dataset, training_config):
 
     print(f"Best model saved to: {best_model_path}")
 if __name__=="__main__":
-    run()
+    train_model()