wehub-resource-sync
13 lines
251 B
Python
13 lines
251 B
Python
from mlflow.models import set_model
|
|
from mlflow.pyfunc import PythonModel
|
|
|
|
|
|
class MyModel(PythonModel):
|
|
def predict(self, context, model_input):
|
|
from utils import my_function
|
|
|
|
return my_function(model_input)
|
|
|
|
|
|
set_model(MyModel())
|