wehub-resource-sync
14 lines
335 B
Python
14 lines
335 B
Python
import mlflow
|
|
|
|
with mlflow.start_run():
|
|
model_info = mlflow.pyfunc.log_model(
|
|
name="model",
|
|
python_model="model.py",
|
|
model_config={"timeout": 10},
|
|
input_example=["hello"],
|
|
)
|
|
|
|
|
|
# model = mlflow.pyfunc.load_model(model_info.model_uri, model_config={"timeout": 10})
|
|
# print(model.predict("hello"))
|