Modify Python script and execution shell script to customize Docker i…

[tanmatth]

…mage URI and rectify issue with input and output S3 locations

What does this PR do?

🛑 Please open an issue first to discuss any significant work and flesh out details/direction - we would hate for your time to be wasted.
Consult the CONTRIBUTING guide for submitting pull-requests.

Motivation

More

• [ X] Yes, I have tested the PR using my local account setup (Provide any test evidence report under Additional Notes)
• [ X] Mandatory for new blueprints. Yes, I have added a example to support my blueprint PR
• Mandatory for new blueprints. Yes, I have updated the website/docs or website/blog section for this feature
• [ X] Yes, I ran pre-commit run -a with this PR. Link for installing pre-commit locally

For Moderators

• E2E Test successfully complete before merge?

Additional Notes

[vara-bonthu]

One comment on the commented xgboost image otherwise it's good.

[vara-bonthu]

LGTM 🚀