label-studio-ml-backend
label-studio-ml-backend copied to clipboard
Docker set up to support reading checkpoint_file from config.json and documentation
Hi,
Created a PR to support passing the model's checkpoint_files in a docker container set up. Here's a sort description of the changes:
- Even though in
label_studio_ml/default_configs/_wsgi.py.tmpl
there is support for reading thecheckpoint_file
throughconfig.json
(see methodget_kwargs_from_config()
) thejson
import is missing as well as making the kwargs available to theuWSGI
app. - Added
config.json
in thedefault_configs
- Updated
Dockerfile
to copy theconfig.json
as well as thecheckpoint_file
that should be placed in./models/
- Updated
README.md
with the instructions on using the backend with docker, including considering the cloud storage credentials and checkpoint file exposure