config.json

{"model_type": "custom_key_class_nn"}