Adds regular python launcher instead of poetry run

README.md

@@ -19,11 +19,11 @@ poetry install
 Once the project has been installed, you can run it with the following command:
 
 ```
-poetry run flashtrain
+python -m hydra_flash.train
 ```
 
 You can, for example, change the learning rate using : 
 
 ```
-poetry run flashtrain model.learning_rate=1e-2
+python -m hydra_flash.train model.learning_rate=1e-2
 ```

hydra_flash/train.py

@@ -23,3 +23,7 @@ def main(cfg):
     predict_datamodule = instantiate(cfg.predict_datamodule)
     predictions = trainer.predict(model, datamodule=predict_datamodule, output="labels")
     log.info(f"predictions : {predictions}")
+
+
+if __name__ == "__main__":
+    main()