wehub-resource-sync
11 lines
224 B
Python
11 lines
224 B
Python
from mlflow.models import set_model
|
|
from mlflow.pyfunc import PythonModel
|
|
|
|
|
|
class MyModel(PythonModel):
|
|
def predict(self, context, model_input):
|
|
return f"This was the input: {model_input}"
|
|
|
|
|
|
set_model(MyModel())
|