in functions/start_model/app.py [0:0]
def lambda_handler(event, context):
lookoutforvision_client = boto3.client("lookoutvision")
project_name = os.environ["lookoutforvision_project_name"]
project_model_version = os.environ["lookoutforvision_project_model_version"]
try:
running_states = ["HOSTED", "STARTING_HOSTING"]
response = lookoutforvision_client.describe_model(
ProjectName=project_name, ModelVersion=project_model_version
)
running_status = response["ModelDescription"]["Status"]
if running_status not in running_states:
response = lookoutforvision_client.start_model(
ProjectName=project_name,
ModelVersion=project_model_version,
MinInferenceUnits=int(
os.environ["minimumInferenceUnitsToUse"]
), # Can be increased upto 5 for running multiple inference units
ClientToken=os.environ["clientToken"],
)
running_status = response["Status"]
print("Current state is: ", running_status)
return running_status
except Exception as e:
print(e)