@Prasanth_Noel This looks pretty good already. Is it not working well?
To be clear, you want to containerize your training/application to run locally on your machine, using LightningWork?
To make sure everything works well, you can take one of the examples from our docs and put it in train.py to verify the app starts correctly.
Note that if you want to run the app/work in the Lightning cloud, it’s even easier since you don’t need a special docker image (usually) and it will install your requirements.txt, Python, pip etc. automatically for you.