1  from mlflow.models import ModelConfig, set_model
 2  
 3  
 4  def predict(model_input):
 5      model_config = ModelConfig()
 6      timeout = model_config.get("timeout")
 7      return [timeout] * len(model_input)
 8  
 9  
10  set_model(predict)