Added click to run the model

Succesfully uploaded model to HF hub in correct place

Uploading models to hub

succesful local run

Refactoring to distangle modules

Refactoring models

Corrected configure_optimizer in lightning module

Initial training code