Colab training Fix: Clean up unused import and update local configuration for Colab training
1 . Removing unused dependencies.
3. Updating local file paths in the configuration for smooth execution in Colab.
4. Ensuring the training script runs successfully with minimal setup.

Please download the below, and run it in Google Colab

Colab_Train.ipynb