chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,12 @@
|
||||
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())
|
||||
Reference in New Issue
Block a user