register_models.py

from azure.ai.ml import MLClient
from azure.ai.ml.entities import Model
from azure.identity import DefaultAzureCredential


def register_model(ml_client, path, name, description):
    model = Model(
        path=path,
        name=name,                 
        description=description
    )

    registered_model = ml_client.models.create_or_update(model)
    print(f"Model registered: {registered_model.name}, version: {registered_model.version}")


def main():
    credential = DefaultAzureCredential()
    ml_client = MLClient(
        credential,
        subscription_id="a846a0af-bfc9-44ae-a421-b5ff06d9e7c5",
        resource_group_name="bps-drone-deploy-dev-rg-uksouth",
        workspace_name="bps-drone-deploy-dev-mlw-uksouth",
    )

    register_model(ml_client, 'checkpoints/building_v4_4_deploy.pth', 'model_building_v4_4', 'Building model')
    register_model(ml_client, 'checkpoints/other_v4_16_deploy.pth', 'model_other_v4_4', 'Other model')


if __name__=='__main__':
    main()