def METHOD_NAME(self, model_horizontal_two_layers):
    geo_data = model_horizontal_two_layers
    # Compute model
    sol = gempy.compute_model(geo_data, compute_mesh_options={'rescale': True})
    return geo_data