Sure Himadri, I'm happy to help.

Did you try using the colab code I provided. Link: https://colab.research.google.com/drive/1b8pVMMoR37a3b9ICo8TMqMLVD-WvbzTk