from mlflow import keras

model = keras.load_model('saved_model.pb')
predict = model.predict([13, 1, 1500, 1500])