Enable selections in docker image build workflow

[HappyAmazonian]

Description

• Enable selections in docker image build by using arch as input arg
• refactor the code to use matrix

• If this change is a backward incompatible change, why must this change be made? N/A

Type of change

Please delete options that are not relevant.

• New feature (non-breaking change which adds functionality)

Checklist:

• Have you manually built the docker image and verify the change?
• Have you run related tests? Check how to set up the test environment here; One example would be pytest tests.py -k "TestCorrectnessLmiDist" -m "lmi_dist"
• Have you added tests that prove your fix is effective or that this feature works?
• Has code been commented, particularly in hard-to-understand areas?
• Have you made corresponding changes to the documentation?

Feature/Issue validation/testing

a run with building all images: https://github.com/deepjavalibrary/djl-serving/actions/runs/12423622213
a run with some images: 12423485871

[siddvenk]

if a user passes in a single container, does it need to be specified like ["lmi"], or will "lmi" work as well?

[HappyAmazonian]

I made a fix so it accepts both now.
https://github.com/deepjavalibrary/djl-serving/actions/runs/12550909958/job/34994447028