problem while serving with docker container

[sawanjr]

The current trainer script in TFX saves the exported model with an additional directory structure before the saved_model.pb file. This structure includes the model version followed by the format ("serving") which creates a path like: model_path/

• version_X/
  • serving/
    • saved_model.pb

However, the Docker container expects the saved_model.pb file to be directly under the model version directory:

model_path/

• version_X/
  • saved_model.pb This mismatch causes issues when loading the model in the Docker environment.