wehub-resource-sync
11 lines
310 B
Python
11 lines
310 B
Python
from mlflow.models import ModelConfig, set_model
|
|
|
|
|
|
def predict(model_input: list[str]):
|
|
model_config = ModelConfig(development_config="tests/pyfunc/sample_code/config.yml")
|
|
timeout = model_config.get("timeout")
|
|
return f"This was the input: {model_input[0]}, timeout {timeout}"
|
|
|
|
|
|
set_model(predict)
|