flask_api.py

from flask import Flask, requests
from ml_model import Model

model = Model()
app = Flask(__name__)

@app.route('/predict', methods=['POST'])
def predict():
  recieved_params = requests.get_json(force=True)
  output = model.predict(recieved_params)
  return output

if __name__ == '__main__':
  app.run()