[UI] Create a model
Context
In #384718 (closed) a model registry was suggested derived from the Package Registry. The model registry is a version control system for models that provides APIs to store and retrieve models and model-related artifacts
Assumed problem
Data scientists are not used to use CLI or CI/CD to populate a model registry. They are more used to using a UI to accomplish this. Currently, it's only possible to add a model to the model registry via ML Flow Client
How might we make the handoff between ML and Ops easier?
Competitors
Competitor | UI to add model | API to add model | UI to add model version | API to add model version |
---|---|---|---|---|
Neptune | ||||
MLflow | ||||
SageMaker | ||||
Verta | ||||
Vertex AI |
Proposal
- Add a UI to add/import/register a model with the model registry.
- Add a UI to add/import/registry a version of an existing model in the model registry.
States:
- Button to add model to model list page, model detail page
- User inputs model name, model version, other things? Description?
- Upload dialog and/or drag and drop
- Uploading progress state
- Error uploading (model already exists, file too large, invalid file type, other errors?...)
- Uploading success state
Designs
SSOT is Figma.
Impression in design management
Considerations
A UI to upload models could be reused as a way to upload packages to the Package Registry
Edited by Alper Akgun