custom model folder for the different versions of your model in the Picsellia model registry

picsellia_model_reg.py

model_name = "detr-resnet-50_"
model_description = "Finetuned DETR model"
if 'api_token' not in os.environ:
    raise Exception("You must set an api_token to run this image")
api_token = os.environ["api_token"]
if "host" not in os.environ:
    host = "https://trial.picsellia.com"
else:
    host = os.environ["host"]
if "organization_name " not in os.environ:
    organization_name = None
else:
    organization_name = os.environ["organization_name"]
client = Client(
    api_token=api_token,
    host=host,
    organization_name=organization_name
)
try:
    my_model = client.create_model(
    name=model_name,
    description=model_description,
    )
except:
     # Get the model that will receive this new version
    my_model = client.get_model(
    name=model_name,
    )