import mlflow
import pandas as pd
model = mlflow.keras.load_model("my_model")
data = pd.read_json('my_model/input_example.json', orient='index')
print(model.predict(data))