def load_model()

in data_preparation/model_utils.py [0:0]

• 5 lines of code
• 1 McCabe index (conditional complexity)

def load_model(model_id="sayakpaul/whitebox-cartoonizer"):
    model_path = snapshot_download(model_id)
    loaded_model = tf.saved_model.load(model_path)
    concrete_func = loaded_model.signatures["serving_default"]
    return concrete_func