metafflow-predictor.py

# Import whatever dependencies your API needs
import os
 
from metaflow import DataArtifact
 
os.environ['METAFLOW_USER'] = 'cortex'
 
class PythonPredictor:
    def __init__(self, config):
        # Initialize your model here
        print("Got config to be %s" % str(config))
        self.model = DataArtifact(config['model_artifact']).data
 
    def predict(self, payload):
        # Payload is the request your endpoint receives
        prediction = self.model
        return prediction