Skip to content

[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